HemanM commited on
Commit
7d6b4cd
·
verified ·
1 Parent(s): 5495249

Create init_save.py

Browse files
Files changed (1) hide show
  1. init_save.py +12 -0
init_save.py ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # init_save.py
2
+
3
+ from evo_model import EvoTransformerForClassification, EvoTransformerConfig
4
+
5
+ # Step 1: Create the config and model
6
+ config = EvoTransformerConfig()
7
+ model = EvoTransformerForClassification(config)
8
+
9
+ # Step 2: Save properly to 'trained_model' directory
10
+ model.save_pretrained("trained_model")
11
+
12
+ print("✅ EvoTransformer base model and config saved to trained_model/")