Archives/petals
2
0
Fork 0
mirror of https://github.com/bigscience-workshop/petals synced 2024-10-31 09:20:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update hivemind to 1.1.2, mark model argument as required

Browse Source
This commit is contained in:
Aleksandr Borzunov 2022-10-25 22:49:37 +00:00
parent 149f433763
commit 7af7671140
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
cli/run_server.py
View File

@ -15,7 +15,7 @@ def main():
parser = configargparse.ArgParser(default_config_files=["config.yml"]) parser = configargparse.ArgParser(default_config_files=["config.yml"])
parser.add('-c', '--config', required=False, is_config_file=True, help='config file path') 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, 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") 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") group.add_argument('model', nargs='?', type=str, help="same as --converted_model_name_or_path")

2
requirements.txt
View File

@ -4,5 +4,5 @@ accelerate==0.10.0
huggingface-hub==0.7.0 huggingface-hub==0.7.0
transformers==4.21.3 transformers==4.21.3
protobuf>=3.12.2,<4.0.0 protobuf>=3.12.2,<4.0.0
hivemind==1.1.1 hivemind==1.1.2
humanfriendly humanfriendly