Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -50,7 +50,7 @@ ckpt_file_streaming_t2v = Path("t2v_enhanced/checkpoints/streaming_t2v.ckpt").ab
|
|
| 50 |
stream_cli, stream_model = init_streamingt2v_model(ckpt_file_streaming_t2v, result_fol)
|
| 51 |
msxl_model = init_v2v_model(cfg_v2v)
|
| 52 |
|
| 53 |
-
|
| 54 |
|
| 55 |
|
| 56 |
# -------------------------
|
|
@@ -68,7 +68,7 @@ def generate(prompt, num_frames, image, model_name_stage1, model_name_stage2, se
|
|
| 68 |
|
| 69 |
n_autoreg_gen = num_frames//8-8
|
| 70 |
|
| 71 |
-
inference_generator.manual_seed(seed)
|
| 72 |
|
| 73 |
if model_name_stage1 == "ModelScopeT2V (text to video)":
|
| 74 |
short_video = ms_short_gen(prompt, ms_model, inference_generator, t, device)
|
|
|
|
| 50 |
stream_cli, stream_model = init_streamingt2v_model(ckpt_file_streaming_t2v, result_fol)
|
| 51 |
msxl_model = init_v2v_model(cfg_v2v)
|
| 52 |
|
| 53 |
+
|
| 54 |
|
| 55 |
|
| 56 |
# -------------------------
|
|
|
|
| 68 |
|
| 69 |
n_autoreg_gen = num_frames//8-8
|
| 70 |
|
| 71 |
+
inference_generator = torch.Generator(device="cuda").manual_seed(seed)
|
| 72 |
|
| 73 |
if model_name_stage1 == "ModelScopeT2V (text to video)":
|
| 74 |
short_video = ms_short_gen(prompt, ms_model, inference_generator, t, device)
|