Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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("/
|
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)
|