Update init_save.py

init_save.py

@@ -4,13 +4,18 @@ import os
 import torch
 from evo_model import EvoTransformerForClassification, EvoTransformerConfig
 
-# ✅ Ensure the directory exists
+# ✅ Create directory if it doesn't exist
 os.makedirs("trained_model", exist_ok=True)
 
-# Create config and model instance
+# ✅ Create config and model
 config = EvoTransformerConfig()
 model = EvoTransformerForClassification(config)
 
-# Save initial model and config
-model.save_pretrained("trained_model")
-print("✅ Initial EvoTransformer saved to trained_model/")
+# ✅ Save like Hugging Face expects
+torch.save(model.state_dict(), "trained_model/pytorch_model.bin")
+
+# Save minimal config
+with open("trained_model/config.json", "w") as f:
+    f.write(config.to_json_string())
+
+print("✅ EvoTransformer initial model saved to trained_model/")