Hugging Face's logo

Spaces:
JackAILab
/
ConsistentID
Running on Zero

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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -51,7 +51,7 @@ pipe.load_ConsistentID_model(
51
  trigger_word="img",
52
  )
53
  pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config)
54
-
55
 
56
  # @torch.inference_mode()
57
  # def Enhance_prompt(prompt,select_images):
 
51
  trigger_word="img",
52
  )
53
  pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config)
54
+ pipe.cuda()
55
 
56
  # @torch.inference_mode()
57
  # def Enhance_prompt(prompt,select_images):