diff --git a/.gitattributes b/.gitattributes
index 74848b6d9c..43a1be3f9f 100644
--- a/.gitattributes
+++ b/.gitattributes
@@ -1 +1,2 @@
 tutorials/* linguist-documentation
+*.pt binary
diff --git a/captum/insights/attr_vis/models/cifar_torchvision.pt b/captum/insights/attr_vis/models/cifar_torchvision.pt
index 41778d7427..ed398a2b23 100644
Binary files a/captum/insights/attr_vis/models/cifar_torchvision.pt and b/captum/insights/attr_vis/models/cifar_torchvision.pt differ
diff --git a/setup.py b/setup.py
index 2fc63e2c16..5726c5c157 100755
--- a/setup.py
+++ b/setup.py
@@ -54,9 +54,9 @@ def report(*args):
 INSIGHTS_REQUIRES = ["flask", "ipython", "ipywidgets", "jupyter"]
 
 INSIGHTS_FILE_SUBDIRS = [
-    "insights/frontend/build",
-    "insights/models",
-    "insights/widget/static",
+    "insights/attr_vis/frontend/build",
+    "insights/attr_vis/models",
+    "insights/attr_vis/widget/static",
 ]
 
 TUTORIALS_REQUIRES = INSIGHTS_REQUIRES + ["torchtext", "torchvision"]