Unverified Commit af388848 authored by Tomek's avatar Tomek Committed by GitHub
Browse files

[example] fixed opt model downloading from huggingface

parent 2ca9728c
......@@ -177,7 +177,7 @@ def main():
default_dist_spec = ShardSpec([-1], [world_size]) if args.shardinit else None
# build model
if args.model_name_or_path is None or args.model_name_or_path == 'facebook/opt-13b':
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])
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment