cluster-summ / configs /model_config.py
jaisidhsingh's picture
Upload model_config.py
58c612d
raw
history blame
226 Bytes
from types import SimpleNamespace
cfg = SimpleNamespace(**{})
# sentence embedding model configs
cfg.sent_model_name = "sentence-transformers/all-MiniLM-L6-v2"
cfg.sent_model_seq_limit = 256
# summarization model configs