System Info
AutoModelForCausalLM.from_pretrained now returns the above error.
Who can help?
No response
Information
Tasks
Reproduction
pip install transformers
from transformers import AutoModelForCausalLM
AutoModelForCausalLM.from_pretrained(
model_path,
cache_dir= cache_dir,
device_map="auto",
quantization_config=bnb_config,
)
Expected behavior
Expected to load the model.
System Info
AutoModelForCausalLM.from_pretrained now returns the above error.
Who can help?
No response
Information
Tasks
examplesfolder (such as GLUE/SQuAD, ...)Reproduction
pip install transformers
from transformers import AutoModelForCausalLM
AutoModelForCausalLM.from_pretrained(
model_path,
cache_dir= cache_dir,
device_map="auto",
quantization_config=bnb_config,
)
Expected behavior
Expected to load the model.