Fix incompiltibility between --revision and --adapter-id (#741)

predibase · Jan 14, 2025 · 73408b5 · 73408b5
1 parent 7195d25
commit 73408b5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/server/lorax_server/cli.py b/server/lorax_server/cli.py
@@ -136,7 +136,8 @@ def download_weights(
     )
     _download_weights(model_id, revision, extension, auto_convert, source, api_token, embedding_dim)
     if adapter_id:
-        _download_weights(adapter_id, revision, extension, auto_convert, adapter_source, api_token)
+        # TODO(travis): allow passing in a revision for the adapter ID
+        _download_weights(adapter_id, None, extension, auto_convert, adapter_source, api_token)
 
 
 @app.command()