Spaces:
Runtime error
Runtime error
fix(src): :rocket: Remove lines
Browse files
app.py
CHANGED
@@ -1,6 +1,5 @@
|
|
1 |
import spaces
|
2 |
import torch
|
3 |
-
# from diffusers import BitsAndBytesConfig
|
4 |
from diffusers import HunyuanVideoPipeline, HunyuanVideoTransformer3DModel
|
5 |
|
6 |
import os
|
@@ -23,10 +22,10 @@ def generate_video(
|
|
23 |
width, height = resolution.split("x")
|
24 |
width, height = int(width), int(height)
|
25 |
|
|
|
26 |
transformer_8bit = HunyuanVideoTransformer3DModel.from_pretrained(
|
27 |
model,
|
28 |
subfolder="transformer",
|
29 |
-
# quantization_config=quant_config,
|
30 |
torch_dtype=torch.bfloat16,
|
31 |
device_map="balanced",
|
32 |
)
|
@@ -62,7 +61,7 @@ def generate_video(
|
|
62 |
|
63 |
return video_path
|
64 |
|
65 |
-
def create_demo(
|
66 |
with gr.Blocks() as demo:
|
67 |
gr.Markdown("# Hunyuan Video Generation")
|
68 |
|
@@ -129,10 +128,5 @@ def create_demo(model):
|
|
129 |
|
130 |
if __name__ == "__main__":
|
131 |
print("Starting Gradio server...")
|
132 |
-
|
133 |
-
# server_name = os.getenv("SERVER_NAME", "0.0.0.0")
|
134 |
-
# server_port = int(os.getenv("SERVER_PORT", "8081"))
|
135 |
-
model = "hunyuanvideo-community/HunyuanVideo" # Actualizado el path del modelo
|
136 |
-
demo = create_demo(model)
|
137 |
-
# demo.launch(server_name=server_name, server_port=server_port)
|
138 |
demo.launch()
|
|
|
1 |
import spaces
|
2 |
import torch
|
|
|
3 |
from diffusers import HunyuanVideoPipeline, HunyuanVideoTransformer3DModel
|
4 |
|
5 |
import os
|
|
|
22 |
width, height = resolution.split("x")
|
23 |
width, height = int(width), int(height)
|
24 |
|
25 |
+
model = "hunyuanvideo-community/HunyuanVideo"
|
26 |
transformer_8bit = HunyuanVideoTransformer3DModel.from_pretrained(
|
27 |
model,
|
28 |
subfolder="transformer",
|
|
|
29 |
torch_dtype=torch.bfloat16,
|
30 |
device_map="balanced",
|
31 |
)
|
|
|
61 |
|
62 |
return video_path
|
63 |
|
64 |
+
def create_demo():
|
65 |
with gr.Blocks() as demo:
|
66 |
gr.Markdown("# Hunyuan Video Generation")
|
67 |
|
|
|
128 |
|
129 |
if __name__ == "__main__":
|
130 |
print("Starting Gradio server...")
|
131 |
+
demo = create_demo()
|
|
|
|
|
|
|
|
|
|
|
132 |
demo.launch()
|