Use --predibase-api-token parameter when downloading (#687)

predibase · Nov 19, 2024 · 1220ce8 · 1220ce8
1 parent f32a11d
commit 1220ce8
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/launcher/src/main.rs b/launcher/src/main.rs
@@ -1148,6 +1148,13 @@ fn download_convert_model(
         envs.push(("HUGGING_FACE_HUB_TOKEN".into(), api_token.into()))
     };
 
+    if let Some(predibase_api_token) = &args.predibase_api_token {
+        envs.push((
+            "PREDIBASE_API_TOKEN".into(),
+            predibase_api_token.to_string().into(),
+        ));
+    }
+
     // If args.weights_cache_override is some, pass it to the download process
     // Useful when running inside a HuggingFace Inference Endpoint
     if let Some(weights_cache_override) = &args.weights_cache_override {