mirror of
https://git.isriupjv.fr/ISRI/ai-server
synced 2025-04-24 18:18:11 +02:00
Models are now loaded in separate endpoints for the inputs to be easier to recognise
15 lines
190 B
JSON
15 lines
190 B
JSON
{
|
|
"type": "python",
|
|
"file": "model.py",
|
|
|
|
"inputs": {
|
|
"prompt": {"type": "str"}
|
|
},
|
|
|
|
"requirements": [
|
|
"transformers",
|
|
"torch",
|
|
"torchvision",
|
|
"torchaudio"
|
|
]
|
|
}
|