File size: 383 Bytes
dbeeca4 25d28df dbeeca4 |
1 2 3 4 5 6 7 8 9 |
from transformers import AutoModelForCausalLM, AutoTokenizer
model_id = "teknium/OpenHermes-2.5-Mistral-7B"
tokenizer = AutoTokenizer.from_pretrained(model_id)
model = AutoModelForCausalLM.from_pretrained(model_id)
# Save to BuddAi/
model.save_pretrained("BuddAi", safe_serialization=True) # Creates model.safetensors
tokenizer.save_pretrained("BuddAi") # Creates tokenizer files |