plano/model_server/app/guard_model_config.json at dd8c43a3927dad49c6865a34cae55df91d1c508a - apunkt/plano - bitfreedom.net: free all bits, everywhere

apunkt/plano

mirror of https://github.com/katanemo/plano.git synced 2026-04-27 09:46:28 +02:00

Co Tran 79b1c5415f

[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>

2024-09-23 12:07:31 -07:00

10 lines

No EOL

239 B

JSON

Raw Blame History

 {
     "toxic":{
         "cpu": "katanemolabs/toxic_ovn_4bit",
         "gpu": "katanemolabs/Bolt-Toxic-v1-eetq"
     },
     "jailbreak":{
         "cpu": "katanemolabs/jailbreak_ovn_4bit",
         "gpu": "katanemolabs/Bolt-Guard-EEtq"
     }
 }