Create init_save.py

init_save.py

@@ -1,12 +1,16 @@
 # init_save.py
 
+import os
+import torch
 from evo_model import EvoTransformerForClassification, EvoTransformerConfig
 
-# Step 1: Create the config and model
+# ✅ Ensure the directory exists
+os.makedirs("trained_model", exist_ok=True)
+
+# Create config and model instance
 config = EvoTransformerConfig()
 model = EvoTransformerForClassification(config)
 
-# Step 2: Save properly to 'trained_model' directory
+# Save initial model and config
 model.save_pretrained("trained_model")
-
-print("✅ EvoTransformer base model and config saved to trained_model/")
+print("✅ Initial EvoTransformer saved to trained_model/")