File size: 286 Bytes
a6cc8ed 5a097b1 a6cc8ed |
1 2 3 4 5 6 7 8 |
# Load model directly
!pip install transformers==4.30.0
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) |