Init

text_to_image.py

@@ -20,7 +20,7 @@ def generate_image(prompt: str) -> torch.Tensor:
     """
     # load model
 
-    pipe = StableDiffusionPipeline.from_pretrained(MODEL_ID, torch_dtype=torch.float16)
+    pipe = StableDiffusionPipeline.from_pretrained(MODEL_ID) #torch_dtype=torch.float16
     pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
     if torch.cuda.is_available():
         pipe = pipe.to("cuda")  # move model to GPU if available