trained_model

# init_save.py

from evo_model import EvoTransformerForClassification, EvoTransformerConfig
import torch

# Create config and model instance
config = EvoTransformerConfig()
model = EvoTransformerForClassification(config)

# Save initial model and config
model.save_pretrained("trained_model")
print("✅ Initial EvoTransformer saved to trained_model/")