helkoo commited on
Commit
34e1d28
·
1 Parent(s): 0fa3320

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ from diffusers.utils import load_image
9
  #Models
10
  openpose = OpenposeDetector.from_pretrained('lllyasviel/ControlNet')
11
  controlnet = ControlNetModel.from_pretrained("lllyasviel/sd-controlnet-openpose", torch_dtype=torch.float16)
12
- pipe = StableDiffusionControlNetPipeline.from_pretrained("/content/drive/MyDrive/hack/800", controlnet=controlnet, safety_checker=None, torch_dtype=torch.float16)
13
 
14
  #optimizations
15
  pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
 
9
  #Models
10
  openpose = OpenposeDetector.from_pretrained('lllyasviel/ControlNet')
11
  controlnet = ControlNetModel.from_pretrained("lllyasviel/sd-controlnet-openpose", torch_dtype=torch.float16)
12
+ pipe = StableDiffusionControlNetPipeline.from_pretrained("helkoo/jelaba_2HR", controlnet=controlnet, safety_checker=None, torch_dtype=torch.float16)
13
 
14
  #optimizations
15
  pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)