hpoghos commited on
Commit
9cd26c4
1 Parent(s): 6be531d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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
- inference_generator = torch.Generator(device="cuda")
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)