Hugging Face's logo

Spaces:
Lam-Hung
/
Interior_design
Runtime error

App Files Community
Lam-Hung commited on
Commit
707baf1
•
1 Parent(s): caed8cb

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
- "runwayml/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
+ "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)