File size: 251 Bytes
a6cc8ed
 
 
 
 
 
1
2
3
4
5
6
# 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)