MindLabUnimib commited on
Commit
a399dba
·
verified ·
1 Parent(s): 447ebcb

Update app_nobatching.py

Browse files
Files changed (1) hide show
  1. app_nobatching.py +3 -3
app_nobatching.py CHANGED
@@ -28,9 +28,9 @@ CLASSIFIER_MODEL_NAME = "saiteki-kai/QA-DeBERTa-v3-large-binary-3"
28
 
29
  # Generation parameters
30
  MAX_NEW_TOKENS = 256
31
- REPETITION_PENALTY = 1.15
32
- MAX_INPUT_LENGTH = 512
33
- MAX_CLASSIFIER_LENGTH = 512 + 256
34
 
35
  # ============================================================================
36
  # Model Loading
 
28
 
29
  # Generation parameters
30
  MAX_NEW_TOKENS = 256
31
+ REPETITION_PENALTY = 1.1
32
+ MAX_INPUT_LENGTH = 256
33
+ MAX_CLASSIFIER_LENGTH = MAX_INPUT_LENGTH + MAX_NEW_TOKENS
34
 
35
  # ============================================================================
36
  # Model Loading