Update pipline_StableDiffusion_ConsistentID.py

pipline_StableDiffusion_ConsistentID.py

@@ -42,6 +42,7 @@ class ConsistentIDStableDiffusionPipeline(StableDiffusionPipeline):
         lora_rank= 128,
         **kwargs,
     ):
+        self.device = "cuda"
         self.lora_rank = lora_rank 
         self.torch_dtype = torch_dtype
         self.num_tokens = num_tokens