Update evo_model.py

evo_model.py

@@ -1,3 +1,4 @@
+import torch
 import torch.nn as nn
 import torch.nn.functional as F
 
@@ -33,4 +34,4 @@ class EvoTransformer(nn.Module):
 
     def forward(self, input_ids):
         x = self.encoder(input_ids)
-        return self.classifier(x)
+        return self.classifier(x)