fix error with import

jshuadvd · Jun 9, 2024 · 63870d6 · 63870d6
1 parent ae683f1
commit 63870d6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train.py b/train.py
@@ -1,5 +1,6 @@
 # Train a LongRoPE model on a given dataset
 # %%
+from src.main import LongRoPEModel
 import torch
 import torch.nn as nn
 import torch.optim as optim
@@ -11,7 +12,6 @@
 import src.main
 
 reload(src.main)
-from src.main import LongRoPEModel
 
 
 # %%