Procházet zdrojové kódy

model name correction

damaoooo před 10 měsíci
rodič
revize
6a886a3740
1 změnil soubory, kde provedl 1 přidání a 1 odebrání
  1. 1 1
      pdf2zh/translator.py

+ 1 - 1
pdf2zh/translator.py

@@ -290,7 +290,7 @@ class OllamaTranslator(BaseTranslator):
                     stream=True,
                 )
                 in_think_block = False
-                is_deepseek_r1 = model == "deepseek-r1"
+                is_deepseek_r1 = "deepseek-r1" in model
                 for chunk in stream:
                     chunk = chunk["message"]["content"]
                     # 只在 deepseek-r1 模型下检查 <think> 块