Update app.py

app.py

@@ -72,7 +72,7 @@ else:
       )
     
 if torch.cuda.is_available():
-  pipe = pipe.to("cuda")
+  pipe = pipe.to("cpu")
   pipe.enable_xformers_memory_efficient_attention()
 
 device = "GPU 🔥" if torch.cuda.is_available() else "CPU 🥶"
@@ -123,7 +123,7 @@ def inference(model_name, prompt, guidance, steps, n_images=1, width=512, height
   if seed == 0:
     seed = random.randint(0, 2147483647)
 
-  generator = torch.Generator('cuda').manual_seed(seed)
+  generator = torch.Generator('cpu').manual_seed(seed)
 
   try:
     if img is not None:
@@ -162,7 +162,7 @@ def txt_to_img(model_path, prompt, n_images, neg_prompt, guidance, steps, width,
           # pipe = current_model.pipe_t2i
 
         if torch.cuda.is_available():
-          pipe = pipe.to("cuda")
+          pipe = pipe.to("cpu")
           pipe.enable_xformers_memory_efficient_attention()
         last_mode = "txt2img"
 
@@ -211,7 +211,7 @@ def img_to_img(model_path, prompt, n_images, neg_prompt, img, strength, guidance
           # pipe = current_model.pipe_i2i
         
         if torch.cuda.is_available():
-          pipe = pipe.to("cuda")
+          pipe = pipe.to("cpu")
           pipe.enable_xformers_memory_efficient_attention()
         last_mode = "img2img"