public_chat / src /config.py
Fazzioni's picture
Update src/config.py
581d21d verified
def get_model_options() -> list[tuple[str, dict]]:
"""Retorna lista de tuplas (label, model_id) para uso em selectbox."""
model_id = 'CEIA-POSITIVO/Qwen4-sft-brainly'
revisions = ['rl-step60','rl-step100','rl-step140','rl-step280']
PRESELECTED_MODELS = [{ 'model_id': model_id,
'revision': r,
'tokenizer': 'Qwen/Qwen3-0.6B',
} for r in revisions]
return [(m['revision'], m) for m in PRESELECTED_MODELS]