setup update to download .pth

deepvk · Apr 22, 2024 · 7a93b90 · 7a93b90
1 parent d8d6ad0
commit 7a93b90
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -111,7 +111,7 @@ def run(self):
 
     package_data={
         # 'tree_ring_watermark': ['src/tree_ring_watermark/open_clip/bpe_simple_vocab_16e6.txt.gz', 'src/tree_ring_watermark/open_clip/model_configs/*.json']
-        "": ["*.gz", "*.json"]
+        "": ["*.gz", "*.json", "*.pth"]
     },
 
     python_requires=">=3.7.0",