モデル軽量化

kaminuma · Dec 9, 2024 · c015ac5 · c015ac5
1 parent 20741b8
commit c015ac5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/app/__init__.py b/app/__init__.py
@@ -21,7 +21,7 @@ def create_app():
     try:
         app.config['NLI_PIPELINE'] = pipeline(
             "zero-shot-classification",
-            model="joeddav/xlm-roberta-large-xnli",
+            model="xlm-roberta-base",
             device=0 if torch.cuda.is_available() else -1  # GPUが利用可能な場合はGPUを使用
         )
         logging.debug("AIモデルの初期化に成功しました。")