Hugging Face's logo

Spaces:
JackAILab
/
ConsistentID
Running on Zero

App Files Community
9
JackAILab commited on
Commit
479696a
·
verified ·
1 Parent(s): 56071bc

Update pipline_StableDiffusion_ConsistentID.py

Browse files
Files changed (1) hide show
  1. pipline_StableDiffusion_ConsistentID.py +0 -1
pipline_StableDiffusion_ConsistentID.py CHANGED
@@ -42,7 +42,6 @@ class ConsistentIDStableDiffusionPipeline(StableDiffusionPipeline):
42
  lora_rank= 128,
43
  **kwargs,
44
  ):
45
- self.device = "cuda"
46
  self.lora_rank = lora_rank
47
  self.torch_dtype = torch_dtype
48
  self.num_tokens = num_tokens
 
42
  lora_rank= 128,
43
  **kwargs,
44
  ):
 
45
  self.lora_rank = lora_rank
46
  self.torch_dtype = torch_dtype
47
  self.num_tokens = num_tokens