-
Notifications
You must be signed in to change notification settings - Fork 6
Open
Description
Similiar to something like this in Python, but do it in Go instead.
from transformers import AutoTokenizer, AutoModelForCausalLM
# Replace with the model you're interested in
model_name = "meta-llama/Llama-3.2-1B"
# Tokenizer and Model loading
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForCausalLM.from_pretrained(model_name)
# Save the model locally
model.save_pretrained("./llama_model")
tokenizer.save_pretrained("./llama_model")
Metadata
Metadata
Assignees
Labels
No labels