Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -11,7 +11,7 @@ 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)
|
| 16 |
pipe.enable_xformers_memory_efficient_attention()
|
| 17 |
pipe.enable_model_cpu_offload()
|
|
|
|
| 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)
|
| 16 |
pipe.enable_xformers_memory_efficient_attention()
|
| 17 |
pipe.enable_model_cpu_offload()
|