You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
lilinwang
changed the title
[Enhancement] Support more Llama2 and Falcon models
[Enhancement] TigerTune: Support more Llama2 and Falcon models
Nov 3, 2023
Currently, the logic tell if a model is Llama2 or Falcon is via hardcoded strings, like "tiiuae/falcon-7b", at
tiger/TigerTune/tigertune/finetuning/llm/text_generation_transformer.py
Line 68 in 8ad8d38
tiger/TigerTune/tigertune/finetuning/llm/text_generation_transformer.py
Line 332 in 8ad8d38
It would be useful to replace these hardcoded strings with a more systematic approach.
The text was updated successfully, but these errors were encountered: