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)
|