ailaw / app.py
Tanu17's picture
Create app.py
9e99c42 verified
Raw
History Blame Contribute Delete
226 Bytes
# Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM
tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-2-7b-hf")
model = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-2-7b-hf")