Unavailability of model
#3
by
ikmeet
- opened
can anyone from team kindly explain this ?
KeyError Traceback (most recent call last)
in <cell line: 2>()
1 from transformers import AutoModelForCausalLM
----> 2 model = AutoModelForCausalLM.from_pretrained("LingoIITGN/ganga-1b", device_map="auto")
2 frames
/usr/local/lib/python3.10/dist-packages/transformers/models/auto/configuration_auto.py in getitem(self, key)
708 return self._extra_content[key]
709 if key not in self._mapping:
--> 710 raise KeyError(key)
711 value = self._mapping[key]
712 module_name = model_type_to_module_name(key)
KeyError: 'mistral'
Can you update your "transformers" package.
pip install transformers -U
aamodthakur
changed discussion status to
closed
okay thanks