RanM commited on
Commit
668182d
·
verified ·
1 Parent(s): d07e1c0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -10,7 +10,9 @@ def load_amused_model():
10
  # pipeline = DiffusionPipeline.from_pretrained("Bakanayatsu/ponyDiffusion-V6-XL-Turbo-DPO")
11
  # AutoPipelineForText2Image.from_pretrained("stabilityai/sd-turbo")
12
  # AutoPipelineForText2Image.from_pretrained("stabilityai/sdxl-turbo")
13
- return DiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4")
 
 
14
 
15
 
16
  # Generate image from prompt using AmusedPipeline
 
10
  # pipeline = DiffusionPipeline.from_pretrained("Bakanayatsu/ponyDiffusion-V6-XL-Turbo-DPO")
11
  # AutoPipelineForText2Image.from_pretrained("stabilityai/sd-turbo")
12
  # AutoPipelineForText2Image.from_pretrained("stabilityai/sdxl-turbo")
13
+ return DiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4",
14
+ safety_checker = None,
15
+ requires_safety_checker = False)
16
 
17
 
18
  # Generate image from prompt using AmusedPipeline