Charles-Elena
/

ControlNet-endpoint-test

stable-diffusion

stable-diffusion-diffusers

endpoints-template

Inference Endpoints

Model card Files Files and versions Community

charleselena commited on Jan 21, 2024

Commit

4fb1dc1

·

verified ·

1 Parent(s): c19e295

commet processor

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -64,7 +64,7 @@ class EndpointHandler():
         self.control_type = "depth"
         self.controlnet = ControlNetModel.from_pretrained(CONTROLNET_MAPPING[self.control_type]["model_id"],torch_dtype=dtype).to(device)
-        processor = AutoProcessor.from_pretrained("CompVis/stable-diffusion-safety-checker")
         # Load StableDiffusionControlNetPipeline

         self.control_type = "depth"
         self.controlnet = ControlNetModel.from_pretrained(CONTROLNET_MAPPING[self.control_type]["model_id"],torch_dtype=dtype).to(device)
+        #processor = AutoProcessor.from_pretrained("CompVis/stable-diffusion-safety-checker")
         # Load StableDiffusionControlNetPipeline