diff --git a/.runpod/tests.json b/.runpod/tests.json new file mode 100644 index 0000000..fb9dfa6 --- /dev/null +++ b/.runpod/tests.json @@ -0,0 +1,24 @@ +{ + "tests": [ + { + "name": "validation_text_input", + "input": { + "prompt": "Hello world" + }, + "timeout": 10000 + } + ], + "config": { + "gpuTypeId": "NVIDIA GeForce RTX 4090", + "gpuCount": 1, + "env": [ + { + "key": "MODEL_NAME", + "value": "phi3" + } + ], + "allowedCudaVersions": [ + "12.8" + ] + } +}