Omnibus commited on
Commit
8055b15
1 Parent(s): bac920b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ for i,model in enumerate(models):
20
  pass
21
  print (loaded_model)
22
 
23
- pipeline = AutoPipelineForImage2Image.from_pretrained("runwayml/stable-diffusion-v1-5", variant="fp16", use_safetensors=True).to("cpu")
24
 
25
  def load_model(model_drop):
26
  pipeline = AutoPipelineForImage2Image.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float32, use_safetensors=True)
 
20
  pass
21
  print (loaded_model)
22
 
23
+ pipeline = AutoPipelineForImage2Image.from_pretrained("runwayml/stable-diffusion-v1-5", safety_checker=None, variant="fp16", use_safetensors=True).to("cpu")
24
 
25
  def load_model(model_drop):
26
  pipeline = AutoPipelineForImage2Image.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float32, use_safetensors=True)