profaker commited on
Commit
c153462
·
verified ·
1 Parent(s): 158689f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ def predict(dict, prompt="", negative_prompt="", guidance_scale=7.5, steps=30, s
30
  add_kwargs["algorithm_type"] = "sde-dpmsolver++"
31
 
32
  scheduler = getattr(diffusers, scheduler_class_name)
33
- pipe.scheduler = scheduler.from_pretrained("SG161222/Realistic_Vision_V5.0_noVAE", subfolder="scheduler", **add_kwargs)
34
 
35
  init_image = dict["image"]
36
  mask_image = dict["mask"]
 
30
  add_kwargs["algorithm_type"] = "sde-dpmsolver++"
31
 
32
  scheduler = getattr(diffusers, scheduler_class_name)
33
+ pipe.scheduler = scheduler.from_pretrained("stablediffusionapi/duchaiten-real3d-nsfw-xl", subfolder="scheduler", **add_kwargs)
34
 
35
  init_image = dict["image"]
36
  mask_image = dict["mask"]