File size: 459 Bytes
472f1d2
127dca6
b976908
472f1d2
 
 
6529956
472f1d2
 
 
7e7e11d
472f1d2
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
model:
  name_or_path: "voxmenthe/modernbert-imdb-sentiment"
  tokenizer_name_or_path: "answerdotai/ModernBERT-base"
  max_length: 880 # 256
  dropout: 0.1
  pooling_strategy: "mean" # Current default, change as needed
  num_weighted_layers: 6 # Match original training config

inference:
  # Default path, can be overridden
  model_path: "voxmenthe/modernbert-imdb-sentiment"
  # Using the same max_length as training for consistency
  max_length: 880 # 256