tiny change to .to()

lib/pipeline_ConsistentID.py

@@ -36,16 +36,16 @@ class ConsistentIDPipeline(StableDiffusionPipeline):
     def to(
         self,
         torch_device: Optional[Union[str, torch.device]] = None,
-        torch_dtype: Optional[torch.dtype] = None,
+        dtype: Optional[torch.dtype] = None,
     ):
-        super().to(torch_device,                dtype=torch_dtype)
-        self.bise_net.to(torch_device,          dtype=torch_dtype)
-        self.clip_encoder.to(torch_device,      dtype=torch_dtype)
-        self.image_proj_model.to(torch_device,  dtype=torch_dtype)
-        self.FacialEncoder.to(torch_device,     dtype=torch_dtype)
+        super().to(torch_device,                dtype=dtype)
+        self.bise_net.to(torch_device,          dtype=dtype)
+        self.clip_encoder.to(torch_device,      dtype=dtype)
+        self.image_proj_model.to(torch_device,  dtype=dtype)
+        self.FacialEncoder.to(torch_device,     dtype=dtype)
         # If the unet is not released, the ip_layers should be moved to the specified device and dtype.
         if not isinstance(self.unet, edict):
-            self.ip_layers.to(torch_device,     dtype=torch_dtype)
+            self.ip_layers.to(torch_device,     dtype=dtype)
         return self
 
     @validate_hf_hub_args