Update dependencyvit.py

Author: fffffgggg54

timm/models/dependencyvit.py

@@ -247,7 +247,7 @@ def get_dependency_mask(self, layers: Optional[Union[List[int], Tuple[int]]] = N
         # L' * [B, N, N]
         # L' * [B, N', N']
         result = []
-        layers = range(len(self.blocks)) if not layers
+        layers = layers if layers else range(len(self.blocks))
         for layer in layers:
             result.append(self.blocks[layer].attn.dependency_mask)
         return result