mirror of https://github.com/hpcaitech/ColossalAI
[example] fixed opt model downloading from huggingface
parent
2ca9728cbb
commit
af3888481d
|
@ -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])
|
||||
|
|
Loading…
Reference in New Issue