[FIX] Half Precision problem

app.py

@@ -10,7 +10,9 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 
 model_id = "stabilityai/stable-diffusion-2-1-base"
 scheduler = EulerDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
-pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, revision="fp16", torch_dtype=torch.float16)
+# [FIX] half precision problem
+# pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, revision="fp16", torch_dtype=torch.float16)
+pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler)
 pipe = pipe.to(device)
 model, preprocess = clip.load("ViT-B/32", device=device)