fix import

delphi-suite · Feb 26, 2024 · 3ec83e2 · 3ec83e2
1 parent 1a5d6a0
commit 3ec83e2
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/delphi/train/__init__.py b/src/delphi/train/__init__.py
@@ -1 +0,0 @@
-from .llama2c.tinystories import Task

diff --git a/src/delphi/train/training_old.py b/src/delphi/train/training_old.py
@@ -28,12 +28,13 @@
 from functools import partial
 
 import torch
-from llama2 import LLaMA2, LLaMA2Args
-from llama2c import Task, model_export
 from torch.distributed import destroy_process_group, init_process_group
 from torch.nn.parallel import DistributedDataParallel as DDP
 from tqdm import tqdm
 
+from delphi.train.llama2 import LLaMA2, LLaMA2Args
+from llama2c import Task, model_export
+
 # -----------------------------------------------------------------------------
 # I/O
 out_dir = "out"