build(utils): only download files for pytorch (hyperonym#167)

DataDropp · Apr 20, 2023 · 10b153f · 10b153f
1 parent a626b95
commit 10b153f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/utils/download.py b/utils/download.py
@@ -19,4 +19,5 @@
         cache_dir=cache_dir,
         local_dir_use_symlinks=False,
         resume_download=True,
+        allow_patterns=["*.bin", "*.json", "*.model", "*.py"],
     )