Update app.py

app.py

@@ -13,8 +13,8 @@ pipe = StableDiffusionPipeline.from_pretrained(model_path, torch_dtype=torch.flo
 # Change Scheduler to DDIMScheduler
 pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
 
-# For memory optimization
-pipe.enable_xformers_memory_efficient_attention()
+# For memory optimization -- only for GPU
+# pipe.enable_xformers_memory_efficient_attention()
 
 # Generate the image from the given text prompt
 def generate_response(prompt):
@@ -30,7 +30,7 @@ def generate_response(prompt):
     width = 512
     
     seed = random.randint(0, 2147483647)
-    generator = torch.Generator(device='cuda').manual_seed(seed)
+    generator = torch.Generator(device='cpu').manual_seed(seed)
     
     with torch.inference_mode():
         imgs = pipe(