From 1e8a4c348282c9d8cb982fb73ec4d5bcb06861db Mon Sep 17 00:00:00 2001
From: orionji <42489349+orionji@users.noreply.github.com>
Date: Thu, 20 Apr 2023 06:28:58 +0000
Subject: [PATCH] build(utils): only download files for pytorch

---
 utils/download.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/utils/download.py b/utils/download.py
index 4bb02d7e..7dbb39f7 100644
--- 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"],
     )