yonikremer commited on
Commit
6007122
1 Parent(s): dfafc18

removed cache dir argument

Browse files
Files changed (1) hide show
  1. download_repo.py +0 -4
download_repo.py CHANGED
@@ -1,5 +1,3 @@
1
- import os
2
-
3
  import urllib3
4
 
5
  from huggingface_hub import snapshot_download
@@ -22,9 +20,7 @@ def download_pytorch_model(name: str) -> None:
22
  """
23
  number_of_seconds_in_a_day: int = 86_400
24
  change_default_timeout(number_of_seconds_in_a_day)
25
- curr_folder: str = os.path.dirname(__file__)
26
  snapshot_download(
27
- cache_dir=os.path.join(curr_folder, "huggingface", "models"),
28
  repo_id=name,
29
  etag_timeout=number_of_seconds_in_a_day,
30
  resume_download=True,
 
 
 
1
  import urllib3
2
 
3
  from huggingface_hub import snapshot_download
 
20
  """
21
  number_of_seconds_in_a_day: int = 86_400
22
  change_default_timeout(number_of_seconds_in_a_day)
 
23
  snapshot_download(
 
24
  repo_id=name,
25
  etag_timeout=number_of_seconds_in_a_day,
26
  resume_download=True,