github/fauxpilot
1
0
Fork 0
mirror of https://github.com/fauxpilot/fauxpilot.git synced 2025-07-06 21:11:53 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

Use os.path.join to get the model name in case the org_name is empty (#222)

Browse source
This commit is contained in:
R0CKSTAR 2024-02-07 21:24:39 +08:00 • committed by GitHub
parent 788edd44f2
commit 65cdc13897
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
python_backend/model.py
View file

@ -1,4 +1,5 @@
import json
import os
import torch
import triton_python_backend_utils as pb_utils
@ -25,7 +26,7 @@ class TritonPythonModel:
def initialize(self, args):
self.model_config = model_config = json.loads(args["model_config"])
org_name = model_config["parameters"].get("org_name", {"string_value": "Salesforce"})["string_value"]
model_name = org_name + "/" + model_config["parameters"]["model_name"]["string_value"]
model_name = os.path.join(org_name, model_config["parameters"]["model_name"]["string_value"])
def get_bool(x):
return model_config["parameters"][x]["string_value"].lower() in ["1", "true"]