Upload folder using huggingface_hub

README.md

@@ -33,7 +33,7 @@ hierarchical_top_k_kernel = flex.triton_hierarchical_sae_loss
 "B -- batch size, K -- top-k, F -- dictionary size, D -- model hidden dim"
 
 loss: torch.Tensor = top_k_kernel(
-      indices: torch.Tensor,  # [B, K]
+    indices: torch.Tensor,  # [B, K]
     weight: torch.Tensor,  # [F, D]
     vals: torch.Tensor,  # [B, K]
     bias: torch.Tensor,  # [D]
@@ -41,7 +41,7 @@ loss: torch.Tensor = top_k_kernel(
 )
 
 loss: torch.Tensor = hierarchical_top_k_kernel(
-      indices: torch.Tensor,  # [B, K]
+    indices: torch.Tensor,  # [B, K]
     weight: torch.Tensor,  # [F, D]
     vals: torch.Tensor,  # [B, K]
     bias: torch.Tensor,  # [D]