Spaces:
Running
Running
model = initialize_model("ckpts")
Browse files
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():
|