Create init_model.py

init_model.py

@@ -0,0 +1,16 @@
+# init_model.py
+
+import os
+import torch
+from evo_model import EvoTransformerForClassification, EvoTransformerConfig
+
+# ✅ Make sure the folder exists
+os.makedirs("trained_model", exist_ok=True)
+
+# ✅ Create config and model
+config = EvoTransformerConfig()
+model = EvoTransformerForClassification(config)
+
+# ✅ Save the model and config in HF format
+model.save_pretrained("trained_model")  # includes config.json + pytorch_model.bin
+print("✅ EvoTransformer initial model saved to 'trained_model/'")