diff --git a/examples/language/opt/train_gemini_opt.py b/examples/language/opt/train_gemini_opt.py index 4ea81ca01..4993ce25d 100755 --- a/examples/language/opt/train_gemini_opt.py +++ b/examples/language/opt/train_gemini_opt.py @@ -178,8 +178,6 @@ def main(): # build model if args.model_name_or_path is None: - # currently, there has a bug in pretrained opt-13b - # we can not import it until huggingface fix it logger.info("Train a new model from scratch", ranks=[0]) with ColoInitContext(device=init_dev, dtype=torch.half, default_dist_spec=default_dist_spec,