NOT-OMEGA commited on
Commit
40c02b8
·
verified ·
1 Parent(s): 3f1a1d6

Update processor_bert.py

Browse files
Files changed (1) hide show
  1. processor_bert.py +1 -1
processor_bert.py CHANGED
@@ -25,7 +25,7 @@ _load_lock = threading.Lock()
25
  MODEL_PATH = os.path.join(os.path.dirname(__file__), 'models', 'log_classifier.joblib')
26
  ONNX_DIR = os.path.join(os.path.dirname(__file__), 'models', 'onnx')
27
  CONFIDENCE_THRESHOLD = 0.30
28
- DEFAULT_BATCH = 64
29
 
30
 
31
  def preload_models():
 
25
  MODEL_PATH = os.path.join(os.path.dirname(__file__), 'models', 'log_classifier.joblib')
26
  ONNX_DIR = os.path.join(os.path.dirname(__file__), 'models', 'onnx')
27
  CONFIDENCE_THRESHOLD = 0.30
28
+ DEFAULT_BATCH = 512
29
 
30
 
31
  def preload_models():