JackAILab commited on
Commit
c841331
·
verified ·
1 Parent(s): 4a027b7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -48,7 +48,7 @@ pipe.load_ConsistentID_model(
48
  subfolder="",
49
  weight_name=os.path.basename(consistentID_path),
50
  trigger_word="img",
51
- ).to(device)
52
  pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config)
53
 
54
 
 
48
  subfolder="",
49
  weight_name=os.path.basename(consistentID_path),
50
  trigger_word="img",
51
+ )
52
  pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config)
53
 
54