diff --git a/examples/deepspeed/deepspeed_autotuning/hf_trainer_api/language_modeling/run_clm.py b/examples/deepspeed/deepspeed_autotuning/hf_trainer_api/language_modeling/run_clm.py index faa8e3699a9..0df1d975653 100644 --- a/examples/deepspeed/deepspeed_autotuning/hf_trainer_api/language_modeling/run_clm.py +++ b/examples/deepspeed/deepspeed_autotuning/hf_trainer_api/language_modeling/run_clm.py @@ -34,12 +34,6 @@ import torch import transformers from datasets import load_dataset -from determined.integrations.huggingface import ( - DetCallback, - create_consistent_hf_args_for_deepspeed, - get_hf_args_with_overwrites, -) -from determined.pytorch import dsat from torch.utils.tensorboard import SummaryWriter from transformers import ( CONFIG_MAPPING, @@ -61,6 +55,7 @@ from transformers.utils.versions import require_version import determined as det +from determined.integrations.huggingface import DetCallback, get_hf_args_with_overwrites from determined.pytorch import dsat # Will error if the minimal version of Transformers is not installed. Remove at your own risks.