소스 검색

add upper and lower limit to countables

kevin 1 년 전
부모
커밋
7bf3d72fae
1개의 변경된 파일2개의 추가작업 그리고 2개의 파일을 삭제
  1. 2 2
      main.py

+ 2 - 2
main.py

@@ -130,9 +130,9 @@ def gpt():
         has_named_params = 'payload' in json_payload
         if 'payload' in json_payload:
             if 'predict_q' in json_payload:
-                predict_q = 5 if json_payload['predict_q'] > 5 else 0 if json_payload['predict_q'] < 0 else json_payload['predict_q']
+                predict_q = 5 if json_payload['predict_q'] > 4 else 0 if json_payload['predict_q'] < 1 else json_payload['predict_q']
             if 'num_choices' in json_payload:
-                num_choices = 5 if json_payload['num_choices'] > 5 else 1 if json_payload['num_choices'] < 1 else json_payload['num_choices']
+                num_choices = 5 if json_payload['num_choices'] > 4 else 1 if json_payload['num_choices'] < 2 else json_payload['num_choices']
             if 'use_video' in json_payload:
                 use_video = json_payload['use_video'] == "1"
             if 'chat_model' in json_payload: