apunkt/plano
1
0
Fork 0
mirror of https://github.com/katanemo/plano.git synced 2026-05-03 12:52:56 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

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

2
model_server/app/commons/constants.py
View file

@ -28,6 +28,6 @@ arch_guard_model_type = {
embedding_model = loader.get_embedding_model()
zero_shot_model = loader.get_zero_shot_model()
prompt_guard_dict = loader.get_prompt_guard(arch_guard_model_type[glb.device])
prompt_guard_dict = loader.get_prompt_guard(arch_guard_model_type[glb.DEVICE])
arch_guard_handler = ArchGuardHanlder(model_dict=prompt_guard_dict)