fix download weights step

predibase · Apr 16, 2024 · 4cc8942 · 4cc8942
1 parent 19fc18e
commit 4cc8942
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/launcher/src/main.rs b/launcher/src/main.rs
@@ -791,6 +791,14 @@ fn download_convert_model(
 ) -> Result<(), LauncherError> {
     // Enter download tracing span
     let _span = tracing::span!(tracing::Level::INFO, "download").entered();
+
+    let adapter_source;
+
+    if let Some(static_adapter_source) = args.static_adapter_source.clone()  {
+        adapter_source = static_adapter_source;
+    } else {
+        adapter_source = args.adapter_source.clone()
+    }
 
     let mut download_args = vec![
         "download-weights".to_string(),
@@ -803,7 +811,7 @@ fn download_convert_model(
         "--source".to_string(),
         args.source.clone(),
         "--adapter-source".to_string(),
-        args.adapter_source.clone(),
+        adapter_source,
     ];
 
     // Model optional revision