imdb-sentiment-demo / config.yaml
voxmenthe's picture
update config with HF model name
7e7e11d
raw
history blame
365 Bytes
model:
name: "voxmenthe/modernbert-imdb-sentiment"
output_dir: "checkpoints"
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