mirror of
https://github.com/bigscience-workshop/petals
synced 2024-10-31 09:20:41 +00:00
Update hivemind to 1.1.2, mark model
argument as required
This commit is contained in:
parent
149f433763
commit
7af7671140
@ -15,7 +15,7 @@ def main():
|
||||
parser = configargparse.ArgParser(default_config_files=["config.yml"])
|
||||
parser.add('-c', '--config', required=False, is_config_file=True, help='config file path')
|
||||
|
||||
group = parser.add_mutually_exclusive_group()
|
||||
group = parser.add_mutually_exclusive_group(required=True)
|
||||
group.add_argument('--converted_model_name_or_path', type=str, default=None,
|
||||
help="path or name of a pretrained model, converted with cli/convert_model.py")
|
||||
group.add_argument('model', nargs='?', type=str, help="same as --converted_model_name_or_path")
|
||||
|
@ -4,5 +4,5 @@ accelerate==0.10.0
|
||||
huggingface-hub==0.7.0
|
||||
transformers==4.21.3
|
||||
protobuf>=3.12.2,<4.0.0
|
||||
hivemind==1.1.1
|
||||
hivemind==1.1.2
|
||||
humanfriendly
|
||||
|
Loading…
Reference in New Issue
Block a user