Skip to content

Commit

Permalink
Merge pull request #100 from Yuvrajsinghspd09/main
Browse files Browse the repository at this point in the history
Fix issue #74: Include model_path in /meta endpoint response
  • Loading branch information
antas-marcin authored Feb 1, 2025
2 parents e6cd4d8 + 677a30b commit 2590553
Showing 1 changed file with 6 additions and 1 deletion.
7 changes: 6 additions & 1 deletion meta.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,7 @@ def __init__(
use_sentence_transformer_vectorizer: bool,
trust_remote_code: bool,
):
self.model_path = model_path
if use_sentence_transformer_vectorizer:
if os.path.exists(f"{model_path}/model_config"):
with open(f"{model_path}/model_config", "r") as f:
Expand All @@ -23,7 +24,11 @@ def __init__(
).to_dict()

def get(self):
return {"model": self.config}
return {"model": self.config,
"model_path": self.model_path,
}



def get_model_type(self):
return self.config["model_type"]
Expand Down

0 comments on commit 2590553

Please sign in to comment.