Spaces:
Running on Zero
Running on Zero
Commit ·
767475e
1
Parent(s): bbe66c1
Fix mmaudio import: add MMAudio dir to sys.path in generate_mmaudio
Browse files
app.py
CHANGED
|
@@ -350,6 +350,12 @@ def generate_taro(video_file, seed_val, cfg_scale, num_steps, mode,
|
|
| 350 |
def generate_mmaudio(video_file, prompt, negative_prompt, seed_val,
|
| 351 |
cfg_strength, num_steps, duration, num_samples):
|
| 352 |
"""MMAudio: flow-matching video-to-audio, 44.1 kHz, 8 s window, text-guided."""
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 353 |
import torchaudio
|
| 354 |
from mmaudio.eval_utils import all_model_cfg, generate, load_video, make_video
|
| 355 |
from mmaudio.model.flow_matching import FlowMatching
|
|
|
|
| 350 |
def generate_mmaudio(video_file, prompt, negative_prompt, seed_val,
|
| 351 |
cfg_strength, num_steps, duration, num_samples):
|
| 352 |
"""MMAudio: flow-matching video-to-audio, 44.1 kHz, 8 s window, text-guided."""
|
| 353 |
+
# MMAudio is a local package in ./MMAudio/ — add it to sys.path so imports work.
|
| 354 |
+
import sys as _sys, os as _os
|
| 355 |
+
_mmaudio_dir = _os.path.join(_os.path.dirname(_os.path.abspath(__file__)), "MMAudio")
|
| 356 |
+
if _mmaudio_dir not in _sys.path:
|
| 357 |
+
_sys.path.insert(0, _mmaudio_dir)
|
| 358 |
+
|
| 359 |
import torchaudio
|
| 360 |
from mmaudio.eval_utils import all_model_cfg, generate, load_video, make_video
|
| 361 |
from mmaudio.model.flow_matching import FlowMatching
|