mirror of
https://github.com/katanemo/plano.git
synced 2026-05-15 11:02:39 +02:00
[Kan-103] add support toxic/jailbreak model (#49)
* add toxic/jailbreak model * fix path loading model * fix syntax * fix bug,lint, format * fix bug * formatting * add parallel + chunking * fix bug * working version * fix onnnx name erorr * device * fix jailbreak config * fix syntax error * format * add requirement + cli download for dockerfile * add task * add skeleton change for envoy filter for prompt guard * fix hardware config * fix bug * add config changes * add gitignore * merge main * integrate arch-guard with filter * add hardware config * nothing * add hardware config feature * fix requirement * fix chat ui * fix onnx * fix lint * remove non intel cpu * remove onnx * working version * modify docker * fix guard time * add nvidia support * remove nvidia * add gpu * add gpu * add gpu support * add gpu support for compose * add gpu support for compose * add gpu support for compose * add gpu support for compose * add gpu support for compose * fix docker file * fix int test * correct gpu docker * upgrad python 10 * fix logits to be gpu compatible * default to cpu dockerfile * resolve comments * fix lint + unused parameters * fix * remove eetq install for cpu * remove deploy gpu --------- Co-authored-by: Adil Hafeez <adil@katanemo.com>
This commit is contained in:
parent
80c554ce1a
commit
79b1c5415f
18 changed files with 1622 additions and 191 deletions
10
model_server/app/guard_model_config.json
Normal file
10
model_server/app/guard_model_config.json
Normal file
|
|
@ -0,0 +1,10 @@
|
|||
{
|
||||
"toxic":{
|
||||
"cpu": "katanemolabs/toxic_ovn_4bit",
|
||||
"gpu": "katanemolabs/Bolt-Toxic-v1-eetq"
|
||||
},
|
||||
"jailbreak":{
|
||||
"cpu": "katanemolabs/jailbreak_ovn_4bit",
|
||||
"gpu": "katanemolabs/Bolt-Guard-EEtq"
|
||||
}
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue