File size: 225 Bytes
e078b1d
 
 
 
 
 
1
2
3
4
5
6
7
from src.models.abstractive import build_generation_config

def test_generation_config_builds():
    hf_name, cfg = build_generation_config("flan_t5_small")
    assert "flan-t5" in hf_name
    assert cfg.max_input_tokens > 0