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

fix lint

Browse source
This commit is contained in:
cotran 2024-10-16 14:54:17 -07:00
parent 10f0a027bf
commit 84ba7d1184
1 changed files with 0 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
model_server/app/loader.py
View file

@ -68,10 +68,8 @@ def get_prompt_guard(model_name):
logger.info("Loading Guard Model...")
if glb.DEVICE == "cpu":
model_class = OVModelForSequenceClassification
else:
model_class = AutoModelForSequenceClassification
prompt_guard = {