a0fe1dfdad
Release / release (push) Waiting to run
* feat: moved config into docs; added banner; auto detect "messages" in input * docs: moved config into docs * chore: added .DS_Store * chore: get the original stuff working again * chore: remove all changes * docs: reduced toc and added small config table --------- Co-authored-by: Tim Pietrusky <tim.pietrusky@runpod.io>
52 lines
1.1 KiB
JSON
52 lines
1.1 KiB
JSON
{
|
|
"tests": [
|
|
{
|
|
"name": "basic_inference_test",
|
|
"input": {
|
|
"prompt": "Write a short poem about artificial intelligence."
|
|
},
|
|
"timeout": 30000
|
|
},
|
|
{
|
|
"name": "openai_messages_test",
|
|
"input": {
|
|
"openai_route": "/v1/chat/completions",
|
|
"openai_input": {
|
|
"messages": [
|
|
{
|
|
"role": "system",
|
|
"content": "You are a helpful assistant that writes concise responses."
|
|
},
|
|
{
|
|
"role": "user",
|
|
"content": "Explain what a neural network is in one sentence."
|
|
}
|
|
],
|
|
"max_tokens": 200,
|
|
"temperature": 0.1
|
|
}
|
|
},
|
|
"timeout": 30000
|
|
}
|
|
],
|
|
"config": {
|
|
"gpuTypeId": "NVIDIA GeForce RTX 4090",
|
|
"gpuCount": 1,
|
|
"env": [
|
|
{
|
|
"key": "MODEL_NAME",
|
|
"value": "HuggingFaceTB/SmolLM2-135M-Instruct"
|
|
}
|
|
],
|
|
"allowedCudaVersions": [
|
|
"12.7",
|
|
"12.6",
|
|
"12.5",
|
|
"12.4",
|
|
"12.3",
|
|
"12.2",
|
|
"12.1"
|
|
]
|
|
}
|
|
}
|