瀏覽代碼

Relax the rest of Hugging Face dependencies (#305)

Alexander Borzunov 2 年之前
父節點
當前提交
98be9ffe4c
共有 2 個文件被更改,包括 5 次插入3 次删除
  1. 2 2
      setup.cfg
  2. 3 1
      src/petals/cli/convert_model.py

+ 2 - 2
setup.cfg

@@ -33,8 +33,8 @@ python_requires = >=3.7
 install_requires =
     torch>=1.12
     bitsandbytes==0.38.0.post2
-    accelerate==0.15.0
-    huggingface-hub==0.11.1
+    accelerate>=0.15.0,<1.0.0
+    huggingface-hub>=0.11.1,<1.0.0
     transformers>=4.25.1,<5.0.0
     speedtest-cli==2.1.3
     hivemind==1.1.7

+ 3 - 1
src/petals/cli/convert_model.py

@@ -6,7 +6,7 @@ import torch.backends.quantized
 import torch.nn as nn
 import transformers
 from hivemind.utils.logging import get_logger
-from huggingface_hub import Repository
+from huggingface_hub import HfApi, Repository
 from tqdm.auto import tqdm
 from transformers.models.bloom.modeling_bloom import BloomModel
 
@@ -66,6 +66,8 @@ def main():
     )
     os.makedirs(args.output_path, exist_ok=True)
 
+    api = HfApi(token=args.use_auth_token)
+    api.create_repo(args.output_repo, repo_type="model", exist_ok=True)
     repo = Repository(args.output_path, clone_from=args.output_repo, use_auth_token=args.use_auth_token)
     repo.git_pull()