Fabrice-TIERCELIN commited on
Commit
28598ee
·
verified ·
1 Parent(s): 9d19cf4
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -43,8 +43,7 @@ def initialize_model(model_path):
43
  print(f"`models_root` exists: {models_root_path}")
44
  hunyuan_video_sampler = HunyuanVideoSampler.from_pretrained(models_root_path, args=args)
45
  print("Model initialized: " + model_path)
46
- #return hunyuan_video_sampler
47
- return None
48
 
49
  model = initialize_model("ckpts")
50
 
@@ -60,7 +59,7 @@ def generate_video(
60
  ):
61
  print("generate_video (prompt: " + prompt + ")")
62
  return generate_video_gpu(
63
- model,
64
  prompt,
65
  resolution,
66
  video_length,
 
43
  print(f"`models_root` exists: {models_root_path}")
44
  hunyuan_video_sampler = HunyuanVideoSampler.from_pretrained(models_root_path, args=args)
45
  print("Model initialized: " + model_path)
46
+ return hunyuan_video_sampler
 
47
 
48
  model = initialize_model("ckpts")
49
 
 
59
  ):
60
  print("generate_video (prompt: " + prompt + ")")
61
  return generate_video_gpu(
62
+ None,
63
  prompt,
64
  resolution,
65
  video_length,