Skip to content

Commit

Permalink
Fix crash when --adapter-id is not set and --adapter-source pbase
Browse files Browse the repository at this point in the history
… is specified (#759)
  • Loading branch information
joseph-predibase authored Feb 11, 2025
1 parent d6072b9 commit 04f38d4
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion server/lorax_server/server.py
Original file line number Diff line number Diff line change
Expand Up @@ -292,7 +292,7 @@ async def serve_inner(
local_url = unix_socket_template.format(uds_path, 0)
server_urls = [local_url]

if adapter_source == PBASE:
if adapter_source == PBASE and adapter_id != "":
logger.info("Got a PBASE adapter source, mapping model ID to S3")
api_token = os.getenv("PREDIBASE_API_TOKEN")
adapter_id = map_pbase_model_id_to_s3(adapter_id, api_token)
Expand Down

0 comments on commit 04f38d4

Please sign in to comment.