Error loading model M2_omni

Unable to recognize "bailing2" model type while loading model with transformers.

`Traceback (most recent call last):
File "test.py", line 10, in
model = AutoModelForCausalLM.from_pretrained(
File "/home/a/.conda/envs/m2-omni/lib/python3.8/site-packages/transformers/models/auto/auto_factory.py", line 526, in from_pretrained
config, kwargs = AutoConfig.from_pretrained(
File "/home/a/.conda/envs/m2-omni/lib/python3.8/site-packages/transformers/models/auto/configuration_auto.py", line 1036, in from_pretrained
raise ValueError(
ValueError: The checkpoint you are trying to load has model type `bailing2` but Transformers does not recognize this architecture. This could be because of an issue with the checkpoint, or because your version of Transformers is out of date.`

Seems to be a problem not registering "bailing2" to the transformers lib