From 84ba7d118440c9f6b16551783c5b23e319fb7213 Mon Sep 17 00:00:00 2001 From: cotran Date: Wed, 16 Oct 2024 14:54:17 -0700 Subject: [PATCH] fix lint --- model_server/app/loader.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/model_server/app/loader.py b/model_server/app/loader.py index 32996f70..2be8777e 100644 --- a/model_server/app/loader.py +++ b/model_server/app/loader.py @@ -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 = {