half precision

model.py

@@ -9,8 +9,8 @@ import numpy as np
 class Model:
     def __init__(self):
         modelID = "runwayml/stable-diffusion-v1-5"
-        #pipeline = StableDiffusionPipeline.from_pretrained(modelID, torch_dtype=torch.float16)
-        self.pipe = StableDiffusionPipeline.from_pretrained(modelID)
+        pipeline = StableDiffusionPipeline.from_pretrained(modelID, torch_dtype=torch.float16)
+        #self.pipe = StableDiffusionPipeline.from_pretrained(modelID)
         #prompt = "a photo of an astronaut riding a horse on mars"
         #n_prompt = "deformed, disfigured"