Skip to content

Commit

Permalink
use class name as string to assist subclassing
Browse files Browse the repository at this point in the history
  • Loading branch information
mattf committed Oct 18, 2024
1 parent 7f3a1cc commit 67b1eab
Showing 1 changed file with 2 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -313,7 +313,8 @@ def __init__(self, **kwargs: Any):
default_hosted_model_name=_DEFAULT_MODEL_NAME,
**({"api_key": api_key} if api_key else {}), # only pass if set
infer_path="{base_url}/chat/completions",
cls=self.__class__.__name__,
# instead of self.__class__.__name__ to assist in subclassing ChatNVIDIA
cls="ChatNVIDIA",
)
# todo: only store the model in one place
# the model may be updated to a newer name during initialization
Expand Down

0 comments on commit 67b1eab

Please sign in to comment.