BoxOfColors commited on
Commit
ebde550
·
1 Parent(s): d5b590c

fix: pre-download BigVGAN vocoder to avoid GPU budget drain on MMAudio cold start

Browse files
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -88,6 +88,12 @@ print("Pre-downloading AudioLDM2 (cvssp/audioldm2)…")
88
  snapshot_download(repo_id="cvssp/audioldm2")
89
  print("AudioLDM2 pre-downloaded.")
90
 
 
 
 
 
 
 
91
  # ================================================================== #
92
  # SHARED CONSTANTS / HELPERS #
93
  # ================================================================== #
 
88
  snapshot_download(repo_id="cvssp/audioldm2")
89
  print("AudioLDM2 pre-downloaded.")
90
 
91
+ # Pre-download MMAudio's BigVGAN vocoder (nvidia/bigvgan_v2_44khz_128band_512x, ~489MB).
92
+ # This is fetched inside the GPU window on cold workers during MMAudio inference/regen.
93
+ print("Pre-downloading BigVGAN vocoder (nvidia/bigvgan_v2_44khz_128band_512x)…")
94
+ snapshot_download(repo_id="nvidia/bigvgan_v2_44khz_128band_512x")
95
+ print("BigVGAN vocoder pre-downloaded.")
96
+
97
  # ================================================================== #
98
  # SHARED CONSTANTS / HELPERS #
99
  # ================================================================== #