Spaces:

JackAILab
/

ConsistentID

Running on Zero

JackAILab commited on May 8

Commit

479696a

•

1 Parent(s): 56071bc

Update pipline_StableDiffusion_ConsistentID.py

Files changed (1) hide show

pipline_StableDiffusion_ConsistentID.py CHANGED Viewed

@@ -42,7 +42,6 @@ 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

         lora_rank= 128,
         **kwargs,
     ):
         self.lora_rank = lora_rank
         self.torch_dtype = torch_dtype
         self.num_tokens = num_tokens