HemanM commited on
Commit
36f3f8d
·
verified ·
1 Parent(s): cae5830

Create trained_model

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