JackAILab commited on
Commit
0688ebe
1 Parent(s): 48713af

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -51,7 +51,12 @@ pipe.load_ConsistentID_model(
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):
 
51
  trigger_word="img",
52
  )
53
  pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config)
54
+
55
+ ### Load to cuda
56
+ pipe.to(device)
57
+ pipe.image_encoder.to(device)
58
+ pipe.image_proj_model.to(device)
59
+ pipe.FacialEncoder.to(device)
60
 
61
  # @torch.inference_mode()
62
  # def Enhance_prompt(prompt,select_images):