karimbenharrak
/

stable-diffusion-xl-refiner-1.0-with-endpoint-handler

StableDiffusionXLImg2ImgPipeline

stable-diffusion

Inference Endpoints

Model card Files Files and versions Community

karimbenharrak commited on Mar 27, 2024

Commit

40e337e

·

verified ·

1 Parent(s): 0821367

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -26,7 +26,7 @@ class EndpointHandler():
                                                  ).to("cuda")
         self.smooth_pipe.enable_model_cpu_offload()
-        self.enable_xformers_memory_efficient_attention()
         self.controlnet = ControlNetModel.from_pretrained(

                                                  ).to("cuda")
         self.smooth_pipe.enable_model_cpu_offload()
+        self.smooth_pipe.enable_xformers_memory_efficient_attention()
         self.controlnet = ControlNetModel.from_pretrained(