Fabrice-TIERCELIN commited on
Commit
29be167
·
verified ·
1 Parent(s): 5d1f182

model = initialize_model("ckpts")

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -45,6 +45,8 @@ def initialize_model(model_path):
45
  print('Model initialized: ' + model_path)
46
  return hunyuan_video_sampler
47
 
 
 
48
  def generate_video(
49
  prompt,
50
  resolution,
@@ -122,8 +124,6 @@ def generate_video_gpu(
122
  return video_path
123
 
124
  def create_demo(model_path):
125
- model = initialize_model(model_path)
126
-
127
  with gr.Blocks() as demo:
128
  if torch.cuda.device_count() == 0:
129
  with gr.Row():
 
45
  print('Model initialized: ' + model_path)
46
  return hunyuan_video_sampler
47
 
48
+ model = initialize_model("ckpts")
49
+
50
  def generate_video(
51
  prompt,
52
  resolution,
 
124
  return video_path
125
 
126
  def create_demo(model_path):
 
 
127
  with gr.Blocks() as demo:
128
  if torch.cuda.device_count() == 0:
129
  with gr.Row():