Update app.py

app.py

@@ -59,7 +59,7 @@ current_model_path = current_model.path
 if is_colab:
   pipe = StableDiffusionPipeline.from_pretrained(
       current_model.path,
-      torch_dtype=torch.float16,
+      torch_dtype=torch.float32,
       scheduler=DPMSolverMultistepScheduler.from_pretrained(current_model.path, subfolder="scheduler"),
       safety_checker=lambda images, clip_input: (images, False)
       )
@@ -67,7 +67,7 @@ if is_colab:
 else:
   pipe = StableDiffusionPipeline.from_pretrained(
       current_model.path,
-      torch_dtype=torch.float16,
+      torch_dtype=torch.float32,
       scheduler=DPMSolverMultistepScheduler.from_pretrained(current_model.path, subfolder="scheduler")
       )
     
@@ -148,14 +148,14 @@ def txt_to_img(model_path, prompt, n_images, neg_prompt, guidance, steps, width,
         if is_colab or current_model == custom_model:
           pipe = StableDiffusionPipeline.from_pretrained(
               current_model_path,
-              torch_dtype=torch.float16,
+              torch_dtype=torch.float32,
               scheduler=DPMSolverMultistepScheduler.from_pretrained(current_model.path, subfolder="scheduler"),
               safety_checker=lambda images, clip_input: (images, False)
               )
         else:
           pipe = StableDiffusionPipeline.from_pretrained(
               current_model_path,
-              torch_dtype=torch.float16,
+              torch_dtype=torch.float32,
               scheduler=DPMSolverMultistepScheduler.from_pretrained(current_model.path, subfolder="scheduler")
               )
           # pipe = pipe.to("cpu")
@@ -197,14 +197,14 @@ def img_to_img(model_path, prompt, n_images, neg_prompt, img, strength, guidance
         if is_colab or current_model == custom_model:
           pipe = StableDiffusionImg2ImgPipeline.from_pretrained(
               current_model_path,
-              torch_dtype=torch.float16,
+              torch_dtype=torch.float32,
               scheduler=DPMSolverMultistepScheduler.from_pretrained(current_model.path, subfolder="scheduler"),
               safety_checker=lambda images, clip_input: (images, False)
               )
         else:
           pipe = StableDiffusionImg2ImgPipeline.from_pretrained(
               current_model_path,
-              torch_dtype=torch.float16,
+              torch_dtype=torch.float32,
               scheduler=DPMSolverMultistepScheduler.from_pretrained(current_model.path, subfolder="scheduler")
               )
           # pipe = pipe.to("cpu")