From 3c0420b7b6d5e5e50ef9f1f440d111c3992402fb Mon Sep 17 00:00:00 2001 From: SvenBrnn Date: Thu, 22 May 2025 04:57:49 +0000 Subject: [PATCH] Fix tets --- .runpod/test.json | 5 ----- .runpod/tests.json | 31 +++++++++++++++++++++++++++++++ 2 files changed, 31 insertions(+), 5 deletions(-) delete mode 100644 .runpod/test.json create mode 100644 .runpod/tests.json diff --git a/.runpod/test.json b/.runpod/test.json deleted file mode 100644 index c6294a9..0000000 --- a/.runpod/test.json +++ /dev/null @@ -1,5 +0,0 @@ -{ - "input": { - "prompt": "How are you?" - } -} \ No newline at end of file diff --git a/.runpod/tests.json b/.runpod/tests.json new file mode 100644 index 0000000..54d27ac --- /dev/null +++ b/.runpod/tests.json @@ -0,0 +1,31 @@ +{ + "tests": [ + { + "name": "validation_text_input", + "input": { + "prompt": "Hello world" + }, + "timeout": 60000 + } + ], + "config": { + "gpuTypeId": "NVIDIA GeForce RTX 4090", + "gpuCount": 1, + "env": [ + { + "key": "MODEL_NAME", + "value": "phi3" + } + ], + "allowedCudaVersions": [ + "12.7", + "12.6", + "12.5", + "12.4", + "12.3", + "12.2", + "12.1", + "12.0" + ] + } +} \ No newline at end of file