Lam-Hung commited on
Commit
ee6d56d
1 Parent(s): 707baf1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -33,7 +33,7 @@ controlnets = [
33
  ]
34
  #load stable diffusion 1.5 and controlnets
35
  pipeline = StableDiffusionControlNetPipeline.from_pretrained(
36
- "botp/stable-diffusion-v1-5", controlnet= controlnets, torch_dtype=torch.float16, use_safetensors=True
37
  )
38
  # take UniPCMultistepScheduler for faster inference
39
  pipeline.scheduler = UniPCMultistepScheduler.from_config(pipeline.scheduler.config)
 
33
  ]
34
  #load stable diffusion 1.5 and controlnets
35
  pipeline = StableDiffusionControlNetPipeline.from_pretrained(
36
+ "SG161222/Realistic_Vision_V5.1_noVAE"", controlnet= controlnets, torch_dtype=torch.float16, use_safetensors=True
37
  )
38
  # take UniPCMultistepScheduler for faster inference
39
  pipeline.scheduler = UniPCMultistepScheduler.from_config(pipeline.scheduler.config)