I can not download the llama2 model with transformers on gcp
I am using vertex ai and workbench and a notebookRAG_llama2_vertexai.ipynbto use llama2 models using transformers:pip install -U git+https://github.com/huggingface/transformers.git git+https://github.com/huggingface/accelerate.gitthen I authenticate HF:from huggingface_hub import notebook_login

# Login to Huggingface to get access to the model
notebook_login()thenimport os
from transformers import AutoTokenizer, AutoModelForCausalLM
import transformers
import torchand when I try to get the model:from transformers import AutoTokenizer, AutoModelForCausalLM
tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-2-7b-chat-hf")

model = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-2-7b-chat-hf")from transformersI getFile Save Error for RAG_llama2_vertexai.ipynb

Invalid response: 524and the kernel crashes…Screen Shot 2024-04-19 at 3.05.09 PM2000×882 111 KBI tried to spin a vm with more memory and it crashes at the same point. Any idea what is going on? or what else I could try?  Many thanks!