Spaces:
Running
Running
File size: 895 Bytes
7d565c7 83028c9 34f85d1 7d565c7 83028c9 4efc6bf 83028c9 4efc6bf 83028c9 34f85d1 83028c9 7d565c7 99f288e 7d565c7 |
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 27 28 29 |
import os
from transformers import BertTokenizer
from evo_model import EvoTransformerConfig, EvoTransformerForClassification
def initialize_and_save_model():
# Step 1: Initialize configuration
config = EvoTransformerConfig()
# Step 2: Initialize model
model = EvoTransformerForClassification(config)
# Step 3: Save model
os.makedirs("trained_model", exist_ok=True)
model.save_pretrained("trained_model")
# Step 4: Save tokenizer (BERT-based)
tokenizer = BertTokenizer.from_pretrained("bert-base-uncased")
tokenizer.save_pretrained("trained_model")
print("✅ EvoTransformer and tokenizer initialized and saved to 'trained_model/'")
def load_model():
model = EvoTransformerForClassification.from_pretrained("trained_model")
return model
# Optional: reinitialize if run directly
if __name__ == "__main__":
initialize_and_save_model()
|