Spaces:
Sleeping
Sleeping
Update retrain_from_feedback.py
Browse files- 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
|
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 |
|