Update app.py

app.py

@@ -20,7 +20,7 @@ for i,model in enumerate(models):
         pass
 print (loaded_model)
 
-pipeline = AutoPipelineForImage2Image.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float32, use_safetensors=True)
+pipeline = AutoPipelineForImage2Image.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float32, variant="fp16", use_safetensors=True)
 
 def load_model(model_drop):
     pipeline = AutoPipelineForImage2Image.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float32, use_safetensors=True)