Fix Format (#61)

stack-info: PR: #61, branch: drisspg/stack/1
pytorch-labs · Oct 21, 2024 · 5475578 · 5475578
1 parent 0e0b44d
commit 5475578
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/attn_gym/utils.py b/attn_gym/utils.py
@@ -10,6 +10,7 @@
     _vmap_for_bhqkv,
     _ModificationType,
 )
+
 # TODO This was moved on nightly, this enables 2.5 and 2.6 | we should remove this once 2.5 is no longer supported
 try:
     from torch._dynamo._trace_wrapped_higher_order_op import TransformGetItemToIndex