HemanM commited on
Commit
4c4f54b
·
verified ·
1 Parent(s): 591a205

Update retrain_from_feedback.py

Browse files
Files changed (1) hide show
  1. retrain_from_feedback.py +1 -1
retrain_from_feedback.py CHANGED
@@ -5,7 +5,7 @@ from torch.utils.data import DataLoader, Dataset
5
  from transformers import AutoTokenizer
6
 
7
  from evo_architecture import mutate_genome, default_config, log_genome
8
- from model import EvoTransformerV22 # Ensure this is compatible with config
9
  import csv
10
  import os
11
 
 
5
  from transformers import AutoTokenizer
6
 
7
  from evo_architecture import mutate_genome, default_config, log_genome
8
+ from evo_model import EvoTransformerV22
9
  import csv
10
  import os
11