Merge pull request FlagAI-Open#375 from csyourui/master

Fix typo in aquila_pretrain.py
jkhomme4 · Jun 12, 2023 · 54aea37 · 54aea37
2 parents 72ed48b + 3395087
commit 54aea37
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/examples/Aquila/Aquila-code/aquila_code_pretrain.py b/examples/Aquila/Aquila-code/aquila_code_pretrain.py
@@ -12,7 +12,7 @@
 from flagai.env_args import EnvArgs
 from flagai.env_trainer_v1 import EnvTrainer
 from flagai.model.aquila_model import AQUILAModel
-from flagai.data.datasets.indexed_dataset.build_index_mappings import _build_train_valid_test_datasets,_build_train_valid_test_weighted_datasets
+from flagai.data.dataset.indexed_dataset.build_index_mappings import _build_train_valid_test_datasets,_build_train_valid_test_weighted_datasets
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 

diff --git a/examples/Aquila/Aquila-pretrain/aquila_pretrain.py b/examples/Aquila/Aquila-pretrain/aquila_pretrain.py
@@ -12,7 +12,7 @@
 from flagai.env_args import EnvArgs
 from flagai.env_trainer_v1 import EnvTrainer
 from flagai.model.aquila_model import AQUILAModel
-from flagai.data.datasets.indexed_dataset.build_index_mappings import _build_train_valid_test_datasets,_build_train_valid_test_weighted_datasets
+from flagai.data.dataset.indexed_dataset.build_index_mappings import _build_train_valid_test_datasets,_build_train_valid_test_weighted_datasets
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 

diff --git a/examples/Aquila/aquila_pretrain.py b/examples/Aquila/aquila_pretrain.py
@@ -12,7 +12,7 @@
 from flagai.env_args import EnvArgs
 from flagai.env_trainer_v1 import EnvTrainer
 from flagai.model.aquila_model import AQUILAModel
-from flagai.data.datasets.indexed_dataset.build_index_mappings import _build_train_valid_test_datasets,_build_train_valid_test_weighted_datasets
+from flagai.data.dataset.indexed_dataset.build_index_mappings import _build_train_valid_test_datasets,_build_train_valid_test_weighted_datasets
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")