init_save.py

from evo_model import EvoTransformerConfig, EvoTransformerForClassification
import os

def initialize_and_save_model(save_path="trained_model"):
    config = EvoTransformerConfig()
    model = EvoTransformerForClassification(config)
    
    os.makedirs(save_path, exist_ok=True)
    model.save_pretrained(save_path)