Fix: Ensure model_id is a string when using a model from s3 (#291)

predibase · Feb 29, 2024 · 54d4d7d · 54d4d7d
1 parent f915df7
commit 54d4d7d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/server/lorax_server/models/__init__.py b/server/lorax_server/models/__init__.py
@@ -116,7 +116,7 @@ def get_model(
             model_path, revision=revision, trust_remote_code=trust_remote_code
         )
         logger.info(f"config_dict: {config_dict}")
-        model_id = model_path
+        model_id = str(model_path)
     elif source == "hub":
         config_dict, _ = PretrainedConfig.get_config_dict(
             model_id, revision=revision, trust_remote_code=trust_remote_code