CaptMetal commited on
Commit
dbeeca4
·
verified ·
1 Parent(s): b3fe6bc

Update tokenizergen.py

Browse files
Files changed (1) hide show
  1. tokenizergen.py +8 -5
tokenizergen.py CHANGED
@@ -1,6 +1,9 @@
1
- from transformers import AutoTokenizer
2
 
3
- # Download and save the tokenizer files
4
- tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-Instruct-v0.2")
5
- tokenizer.save_pretrained("BuddAi") # Creates a folder with all tokenizer files
6
- print("Tokenizer files saved to 'BuddAi/' directory!")
 
 
 
 
1
+ from transformers import AutoModelForCausalLM, AutoTokenizer
2
 
3
+ model_id = "teknium/OpenHermes-2.5-Mistral-7B"
4
+ tokenizer = AutoTokenizer.from_pretrained(model_id)
5
+ model = AutoModelForCausalLM.from_pretrained(model_id)
6
+
7
+ # Save to BuddAi/
8
+ model.save_pretrained("BuddAi", safe_serialization=True) # Creates model.safetensors
9
+ tokenizer.save_pretrained("BuddAi") # Creates tokenizer files