Error loading model M2_omni #33
Description
Unable to recognize "bailing2" model type while loading model with transformers.
Traceback (most recent call last): File "test.py", line 10, in <module> 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