BoxOfColors commited on
Commit
767475e
·
1 Parent(s): bbe66c1

Fix mmaudio import: add MMAudio dir to sys.path in generate_mmaudio

Browse files
Files changed (1) hide show
  1. app.py +6 -0
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