KeyError: 'granite'

#7
by alinmutual - opened

please let me know how can I solve the issue?

/databricks/python/lib/python3.11/site-packages/huggingface_hub/file_download.py:1132: FutureWarning: resume_download is deprecated and will be removed in version 1.0.0. Downloads always resume when possible. If you want to force a new download, use force_download=True.
warnings.warn(
KeyError: 'granite'
File , line 2
1 # drop device_map if running on CPU
----> 2 model = AutoModelForCausalLM.from_pretrained(model_path, device_map=device)

IBM Granite org

Thanks for your interest in our models!

It seems like your Hugging Face transformers package is out of date. Have you tried updating it? It should be at least version 4.45.0.

Sign up or log in to comment