We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 8a18686 commit a7fa4f0Copy full SHA for a7fa4f0
1 file changed
src/maxdiffusion/common_types.py
@@ -81,4 +81,4 @@
81
[CROSS_ATTN_HEAD, None],
82
[CROSS_ATTN_Q_LENGTH, FSDP],
83
[CROSS_ATTN_KV_LENGTH, None],
84
-]
+]
0 commit comments