mirror of
https://github.com/NVIDIA/dgx-spark-playbooks.git
synced 2026-04-25 19:33:53 +00:00
11 lines
261 B
YAML
11 lines
261 B
YAML
|
models:
|
||
|
|
- name: deberta-injection
|
||
|
|
hf_model: deepset/deberta-v3-base-injection
|
||
|
|
device: cuda:0
|
||
|
|
batch_size: 32
|
||
|
|
- name: protectai-injection
|
||
|
|
hf_model: protectai/deberta-v3-base-prompt-injection-v2
|
||
|
|
device: cuda:0
|
||
|
|
batch_size: 32
|
||
|
|
port: 8234
|