Skip to content

Commit 8312acf

Browse files
authored
[feat] Add triton framework for models. (#606)
Add triton framework for models. This PR makes changes consistent with web-tier changes http://go/kaggle-pr/30587. Will discuss with @stevemessick to see if we need to create a new release. FIX=b/353864610 DIFFBASE=http://go/kaggle-pr/30587 Tested - Run ``` api.model_create_new("/tmp/model_meta") api.model_instance_version_create("limakaggle/instance1/triton/var1", "/tmp/empty") ```
1 parent 76bd2db commit 8312acf

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

kaggle/models/model_new_instance_request.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -159,7 +159,7 @@ def framework(self, framework):
159159
"""
160160
if framework is None:
161161
raise ValueError("Invalid value for `framework`, must not be `None`") # noqa: E501
162-
allowed_values = ["tensorFlow1", "tensorFlow2", "tfLite", "tfJs", "pyTorch", "jax", "flax", "pax", "maxText", "gemmaCpp", "tensorRtLlm", "ggml", "gguf", "coral", "scikitLearn", "mxnet", "onnx", "keras", "transformers", "other"] # noqa: E501
162+
allowed_values = ["tensorFlow1", "tensorFlow2", "tfLite", "tfJs", "pyTorch", "jax", "flax", "pax", "maxText", "gemmaCpp", "tensorRtLlm", "ggml", "gguf", "coral", "scikitLearn", "mxnet", "onnx", "keras", "transformers", "triton", "other"] # noqa: E501
163163
if framework not in allowed_values:
164164
raise ValueError(
165165
"Invalid value for `framework` ({0}), must be one of {1}" # noqa: E501

src/KaggleSwagger.yaml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2029,6 +2029,7 @@ definitions:
20292029
- onnx
20302030
- keras
20312031
- transformers
2032+
- triton
20322033
- other
20332034
overview:
20342035
type: string

0 commit comments

Comments
 (0)