Forward HF token to nested model downloads

app.py

@@ -276,16 +276,25 @@ def hub_download_token(token: str | None):
     import stable_audio_3.model_configs as model_configs
 
     original_download = model_configs.hf_hub_download
+    token_env_keys = ("HF_TOKEN", "HUGGING_FACE_HUB_TOKEN")
+    previous_token_env = {key: os.environ.get(key) for key in token_env_keys}
 
     def download_with_user_token(*args, **kwargs):
         kwargs.setdefault("token", token)
         return original_download(*args, **kwargs)
 
     model_configs.hf_hub_download = download_with_user_token
+    for key in token_env_keys:
+        os.environ[key] = token
     try:
         yield
     finally:
         model_configs.hf_hub_download = original_download
+        for key, previous in previous_token_env.items():
+            if previous is None:
+                os.environ.pop(key, None)
+            else:
+                os.environ[key] = previous
 
 
 def generation_preflight_error(