HemanM commited on
Commit
ec179c9
·
verified ·
1 Parent(s): 22d947b

Update retrain_from_feedback.py

Browse files
Files changed (1) hide show
  1. retrain_from_feedback.py +1 -1
retrain_from_feedback.py CHANGED
@@ -62,7 +62,7 @@ def load_feedback():
62
 
63
  # 🔧 Evo model builder from config
64
  def build_model(config):
65
- from model import EvoEncoder
66
  class EvoClassifier(nn.Module):
67
  def __init__(self):
68
  super().__init__()
 
62
 
63
  # 🔧 Evo model builder from config
64
  def build_model(config):
65
+ from evo_model import EvoEncoder
66
  class EvoClassifier(nn.Module):
67
  def __init__(self):
68
  super().__init__()