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