Spaces:
Runtime error
Runtime error
Set this space to GPU config to make it work
Browse files
app.py
CHANGED
|
@@ -25,6 +25,9 @@ hf_hub_download(repo_id="tencent/HunyuanVideo", filename="hunyuan-video-t2v-720p
|
|
| 25 |
hf_hub_download(repo_id="tencent/HunyuanVideo", filename="hunyuan-video-t2v-720p/vae/pytorch_model.pt", local_dir="tencent_HunyuanVideo/hunyuan-video-t2v-720p/vae")
|
| 26 |
|
| 27 |
def initialize_model(model_path):
|
|
|
|
|
|
|
|
|
|
| 28 |
args = parse_args()
|
| 29 |
models_root_path = Path(model_path)
|
| 30 |
if not models_root_path.exists():
|
|
@@ -45,6 +48,10 @@ def generate_video(
|
|
| 45 |
flow_shift,
|
| 46 |
embedded_guidance_scale
|
| 47 |
):
|
|
|
|
|
|
|
|
|
|
|
|
|
| 48 |
seed = None if seed == -1 else seed
|
| 49 |
width, height = resolution.split("x")
|
| 50 |
width, height = int(width), int(height)
|
|
|
|
| 25 |
hf_hub_download(repo_id="tencent/HunyuanVideo", filename="hunyuan-video-t2v-720p/vae/pytorch_model.pt", local_dir="tencent_HunyuanVideo/hunyuan-video-t2v-720p/vae")
|
| 26 |
|
| 27 |
def initialize_model(model_path):
|
| 28 |
+
if torch.cuda.device_count() == 0:
|
| 29 |
+
return None
|
| 30 |
+
|
| 31 |
args = parse_args()
|
| 32 |
models_root_path = Path(model_path)
|
| 33 |
if not models_root_path.exists():
|
|
|
|
| 48 |
flow_shift,
|
| 49 |
embedded_guidance_scale
|
| 50 |
):
|
| 51 |
+
if torch.cuda.device_count() == 0:
|
| 52 |
+
gr.Warning('Set this space to GPU config to make it work.')
|
| 53 |
+
return None
|
| 54 |
+
|
| 55 |
seed = None if seed == -1 else seed
|
| 56 |
width, height = resolution.split("x")
|
| 57 |
width, height = int(width), int(height)
|