init_save.py

# init_save.py

from evo_model import EvoTransformerForClassification, EvoTransformerConfig

# Step 1: Create the config and model
config = EvoTransformerConfig()
model = EvoTransformerForClassification(config)

# Step 2: Save properly to 'trained_model' directory
model.save_pretrained("trained_model")

print("✅ EvoTransformer base model and config saved to trained_model/")