MySpace / app.py
sohamslc5's picture
created app.py
a6cc8ed verified
raw
history blame
251 Bytes
# Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM
tokenizer = AutoTokenizer.from_pretrained("sohamslc5/llama_2_7B_sohamslc5")
model = AutoModelForCausalLM.from_pretrained("sohamslc5/llama_2_7B_sohamslc5")
print(model)