heheyas commited on
Commit
63d4833
1 Parent(s): b1531dc

update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -2
app.py CHANGED
@@ -187,9 +187,14 @@ ae_model.load_state_dict(encoder_sd)
187
  ae_model = ae_model.to(device)
188
  rembg_session = rembg.new_session()
189
 
190
- model_config.model.params.ckpt_path = V3D_ckpt_path
191
  model, _ = load_model(
192
- model_config, device, num_frames, num_steps, min_cfg=3.5, max_cfg=3.5
 
 
 
 
 
 
193
  )
194
  model = model.to(device)
195
 
 
187
  ae_model = ae_model.to(device)
188
  rembg_session = rembg.new_session()
189
 
 
190
  model, _ = load_model(
191
+ model_config,
192
+ device,
193
+ num_frames,
194
+ num_steps,
195
+ min_cfg=3.5,
196
+ max_cfg=3.5,
197
+ ckpt_path=V3D_ckpt_path,
198
  )
199
  model = model.to(device)
200