Fixed a bug where TorchLens wasn't accounting for the dim argument in…

… torch.max and torch.min (they return a tuple subclass with the values and the indices).
johnmarktaylor91 · Aug 22, 2024 · 2a25882 · 2a25882
1 parent 19d6556
commit 2a25882
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -14,7 +14,7 @@
 
 setup(
     name="torchlens",
-    version="0.1.17",
+    version="0.1.18",
     description="A package for extracting activations from PyTorch models",
     long_description="A package for extracting activations from PyTorch models. Contains functionality for "
                      "extracting model activations, visualizing a model's computational graph, and "