Error in the model name
I am testing in vscode.
In this part of the code I get an error.
model_id = "refuelai/Llama-3-Refueled".
This is the error generated, in the model name.
Traceback (most recent call last):
File "f:\DevelopmentDataindex3.py", line 6, in .
model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.float16, device_map="auto")
File "F:\F:\Development\Data\data\libsite-packages\transformers\models\auto\auto_factory.py", line 441, in from_pretrained
config, kwargs = AutoConfig.from_pretrained(
File "F:\F:\Data_data\libsite-packages\transformers\models\autoconfiguration_auto.py", line 917, in from_pretrained
config_class = CONFIG_MAPPING[config_dict["model_type"]]
File "F:__F:__Data_data_auto.py", line 623, in getitem.
raise KeyError(key)
KeyError: 'llama'.
Any suggestions?