Skip to content

Commit f5ff1a1

Browse files
committed
update code
1 parent 04c2f3c commit f5ff1a1

File tree

2 files changed

+3
-4
lines changed

2 files changed

+3
-4
lines changed

fastdeploy/input/ernie_vl_processor.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -165,6 +165,7 @@ def _check_mm_limits(self, item):
165165
def process_request_dict(self, request, max_model_len=None):
166166
"""process the input data"""
167167

168+
request['enable_thinking'] = self.get_enable_thinking(request.get("enable_thinking"))
168169
if not request.get("eos_token_ids"):
169170
request["eos_token_ids"] = self.eos_token_ids
170171

@@ -250,7 +251,7 @@ def process_response_dict(self, response_dict, stream, **kwargs):
250251
Returns:
251252
Dict: response contain text fields
252253
"""
253-
enable_thinking = self.get_enable_thinking(kwargs.get("enable_thinking"))
254+
enable_thinking = self.get_enable_thinking(kwargs.pop("enable_thinking", None))
254255
if stream:
255256
return self.process_response_dict_streaming(response_dict, enable_thinking=enable_thinking, **kwargs)
256257
else:

fastdeploy/input/text_processor.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -445,9 +445,7 @@ def process_response_dict(self, response_dict, **kwargs):
445445
Returns:
446446
Dict: response contain text fields
447447
"""
448-
enable_thinking = kwargs.pop("enable_thinking", True)
449-
if enable_thinking is None:
450-
enable_thinking = True
448+
enable_thinking = self.get_enable_thinking(kwargs.pop("enable_thinking", None))
451449
stream = kwargs.get("stream", True)
452450
if stream:
453451
return self.process_response_dict_streaming(

0 commit comments

Comments
 (0)