Currently, the logic tell if a model is Llama2 or Falcon is via hardcoded strings, like "tiiuae/falcon-7b", at

It would be useful to replace these hardcoded strings with a more systematic approach.