Torch 16

image_transformation.py

@@ -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.controlnet = ControlNetModel.from_pretrained(self.controlnet_checkpoint)
         self.pipeline = StableDiffusionControlNetPipeline.from_pretrained(
             self.stable_diffusion_checkpoint, controlnet=self.controlnet, torch_dtype=torch.float16
         )