Update app.py

app.py

@@ -67,12 +67,14 @@ def Generate(image_input, prompt, negative_prompt, strength, guidance_scale, num
 	return image, f"{minutes:02d}:{seconds:02d}"
 def Loading(model):
 	global text2img, img2img
-	text2img = StableDiffusionPipeline.from_pretrained(model, torch_dtype=torch.float16, use_safetensors=True).to(device)
-	text2img.safety_checker = None
-	text2img.scheduler = EulerDiscreteScheduler.from_config(text2img.scheduler.config)
 	if device == "cuda":
+		text2img = StableDiffusionPipeline.from_pretrained(model, torch_dtype=torch.float16, variant="fp16", use_safetensors=True).to(device)
 		text2img.enable_xformers_memory_efficient_attention()
 		text2img.vae.enable_xformers_memory_efficient_attention()
+	else:
+		text2img = StableDiffusionPipeline.from_pretrained(model, use_safetensors=True).to(device)
+	text2img.safety_checker = None
+	text2img.scheduler = EulerDiscreteScheduler.from_config(text2img.scheduler.config)
 	img2img = StableDiffusionImg2ImgPipeline(**text2img.components)
 	return model
 with gr.Blocks() as demo: