소스 검색

Merge pull request #9082 from Alex1607/main

FIX max_tokens not being set properly
Timothy Jaeryang Baek 3 달 전
부모
커밋
40a4443949
1개의 변경된 파일3개의 추가작업 그리고 0개의 파일을 삭제
  1. 3 0
      backend/open_webui/utils/middleware.py

+ 3 - 0
backend/open_webui/utils/middleware.py

@@ -666,6 +666,9 @@ def apply_params_to_form_data(form_data, model):
         if "temperature" in params:
             form_data["temperature"] = params["temperature"]
 
+        if "max_tokens" in params:
+            form_data["max_tokens"] = params["max_tokens"]
+
         if "top_p" in params:
             form_data["top_p"] = params["top_p"]