Spaces:
Runtime error
Runtime error
heheyas
commited on
Commit
·
63d4833
1
Parent(s):
b1531dc
update app.py
Browse files
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,
|
|
|
|
|
|
|
|
|
|
|
|
|
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 |
|