From 3a095f3e10af6a1e2ccd9bf526f9bd117b986a77 Mon Sep 17 00:00:00 2001 From: Ezekiel Wotring <40004347+KAJdev@users.noreply.github.com> Date: Thu, 3 Apr 2025 09:10:01 -0800 Subject: [PATCH] Update tests.json --- .runpod/tests.json | 82 +++++++++++++++++++++++----------------------- 1 file changed, 41 insertions(+), 41 deletions(-) diff --git a/.runpod/tests.json b/.runpod/tests.json index ac00049..5391182 100644 --- a/.runpod/tests.json +++ b/.runpod/tests.json @@ -1,44 +1,44 @@ { - "tests": [ - { - "name": "basic_inference_test", - "input": { - "prompt": "Write a short poem about artificial intelligence.", - }, - "timeout": 30000 - } - ], - "config": { - "gpuTypeId": "NVIDIA GeForce RTX 4090", - "gpuCount": 1, - "env": [ - { - "key": "MODEL_NAME", - "value": "facebook/opt-350m" - }, - { - "key": "HF_TOKEN", - "value": "hf_dummy_token_for_testing_purposes_only" - }, - { - "key": "MAX_MODEL_LEN", - "value": "8192" - }, - { - "key": "GPU_MEMORY_UTILIZATION", - "value": "0.95" - } + "tests": [ + { + "name": "basic_inference_test", + "input": { + "prompt": "Write a short poem about artificial intelligence." + }, + "timeout": 30000 + } ], - "allowedCudaVersions": [ - "12.7", - "12.6", - "12.5", - "12.4", - "12.3", - "12.2", - "12.1", - "12.0", - "11.7" - ] - } + "config": { + "gpuTypeId": "NVIDIA GeForce RTX 4090", + "gpuCount": 1, + "env": [ + { + "key": "MODEL_NAME", + "value": "facebook/opt-350m" + }, + { + "key": "HF_TOKEN", + "value": "hf_dummy_token_for_testing_purposes_only" + }, + { + "key": "MAX_MODEL_LEN", + "value": "8192" + }, + { + "key": "GPU_MEMORY_UTILIZATION", + "value": "0.95" + } + ], + "allowedCudaVersions": [ + "12.7", + "12.6", + "12.5", + "12.4", + "12.3", + "12.2", + "12.1", + "12.0", + "11.7" + ] + } }