Ver Fonte

Update hivemind to 1.1.2, mark `model` argument as required

Aleksandr Borzunov há 2 anos atrás
pai
commit
7af7671140
2 ficheiros alterados com 2 adições e 2 exclusões
  1. 1 1
      cli/run_server.py
  2. 1 1
      requirements.txt

+ 1 - 1
cli/run_server.py

@@ -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")

+ 1 - 1
requirements.txt

@@ -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