[Chatllama] facebook/opt-350m is missing in rlhf/model_list.py

[tengxiaoliu]

Hi, thanks for the great repo!

In rlhf/model_list.py, "facebook/opt-350m" is missing in hf_models_causal_lm. Setting "model=facebook/opt-350m" in actor_config will cause "UnboundLocalError: local variable 'tokenizer' referenced before assignment" at rlhf/actor.py:176.

Or maybe simply add a warning for unsupported models when loading tokenizer at Line176 in chatllama/rlhf/actor.py.

[PierpaoloSorbellini]

Hi @tengxiaoliu, thanks for reaching out and for the advice.
We did miss the 350m version when adding the strings to model_list.py.
We have added opt-350m and an error message for unsupported models. (in PR #306)
Thanks again for the feedback!

[tengxiaoliu]

Thanks for your reply! I'll close this issue.