File size: 393 Bytes
0ab67ad 226aae3 0ab67ad 226aae3 0ab67ad 226aae3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
# save_model.py
from model.luna_model import LunaAI
from transformers import BertTokenizer
def save_model(model, path='./luna_ai_model'):
model.save_pretrained(path)
tokenizer = BertTokenizer.from_pretrained('bert-base-uncased')
tokenizer.save_pretrained(path)
if __name__ == "__main__":
model = LunaAI(num_classes=2) # Adjust num_classes if necessary
save_model(model)
|