CaptMetal commited on
Commit
2595b8d
·
verified ·
1 Parent(s): 2e657ac

Update model.safetensors

Browse files
Files changed (1) hide show
  1. model.safetensors +4 -3
model.safetensors CHANGED
@@ -1,8 +1,9 @@
1
  from transformers import AutoModelForCausalLM, AutoTokenizer
2
 
3
- # Load and save Mistral components (consistent architecture)
4
- model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-Instruct-v0.2")
5
- tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-Instruct-v0.2")
6
 
 
7
  model.save_pretrained("BuddAi", safe_serialization=True) # Creates model.safetensors
8
  tokenizer.save_pretrained("BuddAi") # Creates tokenizer files
 
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