imdb-sentiment-demo / config.yaml
voxmenthe's picture
update inference to load from hf model id
b976908
raw
history blame
393 Bytes
model:
name: "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
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