Fix Format (#61)

drisspg · web-flow · commit 54755782172e · 2024-10-21T16:26:46.000-07:00
stack-info: PR: #61, branch: drisspg/stack/1
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

Original file line number	Diff line number	Diff line change
`@@ -10,6 +10,7 @@`
`10`	`10`	`_vmap_for_bhqkv,`
`11`	`11`	`_ModificationType,`
`12`	`12`	`)`
	`13`	`+`
`13`	`14`	`# TODO This was moved on nightly, this enables 2.5 and 2.6 \| we should remove this once 2.5 is no longer supported`
`14`	`15`	`try:`
`15`	`16`	`from torch._dynamo._trace_wrapped_higher_order_op import TransformGetItemToIndex`