mart9992
/

theimageconvert2

Model card Files Files and versions

mart9992 commited on Jan 7, 2024

Commit

0e959bc

·

1 Parent(s): 1657788

Update entklei.py

Files changed (1) hide show

entklei.py +1 -1

entklei.py CHANGED Viewed

@@ -127,7 +127,7 @@ base_pipe = StableDiffusionControlNetInpaintPipeline.from_pretrained(
 )
 base_pipe.scheduler = DDIMScheduler.from_config(base_pipe.scheduler.config)
 base_pipe = base_pipe.to("cuda")
-base_pipe.enable_model_cpu_offload()
 base_pipe.safety_checker = None
 # base_pipe.enable_xformers_memory_efficient_attention()

 )
 base_pipe.scheduler = DDIMScheduler.from_config(base_pipe.scheduler.config)
 base_pipe = base_pipe.to("cuda")
+# base_pipe.enable_model_cpu_offload()
 base_pipe.safety_checker = None
 # base_pipe.enable_xformers_memory_efficient_attention()