Browse Source

[example] fix redundant note (#3065)

pull/3072/head
binmakeswell 2 years ago committed by GitHub
parent
commit
360674283d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 2
      examples/language/opt/train_gemini_opt.py

2
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,

Loading…
Cancel
Save