helkoo commited on
Commit
5eee034
·
1 Parent(s): d0a5669

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -12,7 +12,6 @@ controlnet = ControlNetModel.from_pretrained("lllyasviel/sd-controlnet-openpose"
12
  pipe = StableDiffusionControlNetPipeline.from_pretrained("helkoo/jelaba_2HR", controlnet=controlnet, safety_checker=None, torch_dtype=torch.float16)
13
 
14
  #optimizations
15
- CUDA_VISIBLE_DEVICES=1
16
  pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
17
  #pipe.enable_xformers_memory_efficient_attention()
18
  #pipe.enable_model_cpu_offload()
 
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()