apunkt/plano
1
0
Fork 0
mirror of https://github.com/katanemo/plano.git synced 2026-05-21 13:55:15 +02:00
Code Issues Projects Releases Packages Wiki Activity

add test and pre commit hook

Browse source
This commit is contained in:
cotran 2024-10-16 14:15:13 -07:00
parent cf612409f0
commit ee36e403d1
9 changed files with 351 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

1
model_server/app/prompt_guard/model_handler.py
View file

@ -11,7 +11,6 @@ class ArchGuardHanlder:
self.model = model_dict["model"]
self.tokenizer = model_dict["tokenizer"]
self.device = model_dict["device"]
self.hardware_config = model_dict["hardware_config"]
self.threshold = threshold