Hugging Face's logo

mart9992
/
theimageconvert2

ONNX
Inference Endpoints
Model card Files Community
mart9992 commited on
Commit
0e959bc
•
1 Parent(s): 1657788

Update entklei.py

Browse files
Files changed (1) hide show
  1. entklei.py +1 -1
entklei.py CHANGED
@@ -127,7 +127,7 @@ base_pipe = StableDiffusionControlNetInpaintPipeline.from_pretrained(
127
  )
128
  base_pipe.scheduler = DDIMScheduler.from_config(base_pipe.scheduler.config)
129
  base_pipe = base_pipe.to("cuda")
130
- base_pipe.enable_model_cpu_offload()
131
  base_pipe.safety_checker = None
132
  # base_pipe.enable_xformers_memory_efficient_attention()
133
 
 
127
  )
128
  base_pipe.scheduler = DDIMScheduler.from_config(base_pipe.scheduler.config)
129
  base_pipe = base_pipe.to("cuda")
130
+ # base_pipe.enable_model_cpu_offload()
131
  base_pipe.safety_checker = None
132
  # base_pipe.enable_xformers_memory_efficient_attention()
133