File size: 766 Bytes
4ca2e6f
 
86c4fc8
2f1948a
4ca2e6f
 
d82cf78
4ca2e6f
 
 
 
2f1948a
4ca2e6f
 
 
 
90aa9f3
4ca2e6f
 
 
 
 
 
 
2f1948a
4ca2e6f
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
from transformers import AutoModelForCausalLM, AutoTokenizer
from huggingface_hub import HfApi, login
import os

# 1. Login (add HF_TOKEN in Space Settings -> Secrets)
login(token=os.environ["HF_TOKEN"])

# 2. Load and save model files
model_id = "teknium/OpenHermes-2.5-Mistral-7B" 
tokenizer = AutoTokenizer.from_pretrained(model_id)
model = AutoModelForCausalLM.from_pretrained(model_id)

# Save files temporarily
os.makedirs("BuddAi", exist_ok=True)
model.save_pretrained("BuddAi", safe_serialization=True)
tokenizer.save_pretrained("BuddAi")

# 3. Upload to your model repo
api = HfApi()
api.upload_folder(
    folder_path="BuddAi",
    repo_id="CaptMetal/BuddAi",
    repo_type="model"
)

print("Upload complete! Visit https://huggingface.co/CaptMetal/BuddAi")