Skip to content

Commit 2bcb887

Browse files
authored
Add nvidia-l40s as a valid TRT compilation target in inference-cli (#2121)
* allow nvidia-l40s as trt compilation target (will run on crusoe) * update docs
1 parent 9eac874 commit 2bcb887

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

  • inference_cli/lib/roboflow_cloud/batch_processing

inference_cli/lib/roboflow_cloud/batch_processing/entities.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -159,6 +159,7 @@ class WorkflowProcessingJobV1(BaseModel):
159159
class CompilationDevice(str, Enum):
160160
NVIDIA_L4 = "nvidia-l4"
161161
NVIDIA_T4 = "nvidia-t4"
162+
NVIDIA_L40S = "nvidia-l40s"
162163

163164

164165
class TRTCompilationJobV1(BaseModel):

0 commit comments

Comments
 (0)