Spaces:

huggingface-tools
/

image-transformation

Running

lysandre HF staff commited on May 3, 2023

Commit

b14f00e

1 Parent(s): e5e5100

Torch 16

Files changed (1) hide show

image_transformation.py CHANGED Viewed

@@ -64,7 +64,7 @@ class ImageTransformationTool(Tool):
         self.controlnet = ControlNetModel.from_pretrained(self.controlnet_checkpoint)
         self.pipeline = StableDiffusionControlNetPipeline.from_pretrained(
-            self.stable_diffusion_checkpoint, controlnet=self.controlnet, torch_dtype=torch.float16
         )
         self.pipeline.scheduler = UniPCMultistepScheduler.from_config(self.pipeline.scheduler.config)
         self.pipeline.enable_model_cpu_offload()

         self.controlnet = ControlNetModel.from_pretrained(self.controlnet_checkpoint)
         self.pipeline = StableDiffusionControlNetPipeline.from_pretrained(
+            self.stable_diffusion_checkpoint, controlnet=self.controlnet
         )
         self.pipeline.scheduler = UniPCMultistepScheduler.from_config(self.pipeline.scheduler.config)
         self.pipeline.enable_model_cpu_offload()