Update generate_image.py

generate_image.py

@@ -4,11 +4,10 @@ from PIL import Image
 import os
 
 pipe = StableDiffusionXLPipeline.from_pretrained(
-    "stabilityai/sdxl-turbo",
-    torch_dtype=torch.float16,
-    variant="fp16"
-).to("cuda" if torch.cuda.is_available() else "cpu")
-pipe.safety_checker = None
+    "stabilityai/sdxl-turbo"
+).to("cpu")
+
+pipe.safety_checker = None  # Optional
 
 def generate_scene_image(prompt, idx):
     image = pipe(prompt).images[0]