瀏覽代碼

Display thinking for deepseek r1 (#4904)

### What problem does this PR solve?
#4903
### Type of change

- [x] New Feature (non-breaking change which adds functionality)
tags/v0.17.0
Kevin Hu 8 月之前
父節點
當前提交
b08bb56f6c
No account linked to committer's email address
共有 2 個文件被更改,包括 8 次插入2 次删除
  1. 1
    1
      deepdoc/vision/operators.py
  2. 7
    1
      rag/llm/chat_model.py

+ 1
- 1
deepdoc/vision/operators.py 查看文件

@@ -108,7 +108,7 @@ class NormalizeImage(object):

def __init__(self, scale=None, mean=None, std=None, order='chw', **kwargs):
if isinstance(scale, str):
scale = np.float32(scale) if scale != 'None' else None
scale = eval(scale)
self.scale = np.float32(scale if scale is not None else 1.0 / 255.0)
mean = mean if mean is not None else [0.485, 0.456, 0.406]
std = std if std is not None else [0.229, 0.224, 0.225]

+ 7
- 1
rag/llm/chat_model.py 查看文件

@@ -73,7 +73,13 @@ class Base(ABC):
continue
if not resp.choices[0].delta.content:
resp.choices[0].delta.content = ""
ans += resp.choices[0].delta.content
if hasattr(resp.choices[0].delta, "reasoning_content") and resp.choices[0].delta.reasoning_content:
if ans.find("<think>") < 0:
ans += "<think>"
ans = ans.replace("</think>", "")
ans += resp.choices[0].delta.reasoning_content + "</think>"
else:
ans += resp.choices[0].delta.content

tol = self.total_token_count(resp)
if not tol:

Loading…
取消
儲存