Spaces:

huggingface-tools
/

image-transformation

Running

lysandre HF Staff commited on May 3, 2023

Commit

e5e5100

1 Parent(s): 5cbbc02

Torch 16

Files changed (1) hide show

image_transformation.py CHANGED Viewed

@@ -62,7 +62,7 @@ class ImageTransformationTool(Tool):
         if self.device is None:
             self.device = get_default_device()
-        self.controlnet = ControlNetModel.from_pretrained(self.controlnet_checkpoint, torch_dtype=torch.float16)
         self.pipeline = StableDiffusionControlNetPipeline.from_pretrained(
             self.stable_diffusion_checkpoint, controlnet=self.controlnet, torch_dtype=torch.float16
         )

         if self.device is None:
             self.device = get_default_device()
+        self.controlnet = ControlNetModel.from_pretrained(self.controlnet_checkpoint)
         self.pipeline = StableDiffusionControlNetPipeline.from_pretrained(
             self.stable_diffusion_checkpoint, controlnet=self.controlnet, torch_dtype=torch.float16
         )