Skip to content

Commit ada408b

Browse files
committed
shifting to dot product attn for time being
1 parent 9b4a16c commit ada408b

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

src/maxdiffusion/configs/ltx2_video.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
#hardware
22
hardware: 'tpu'
33
skip_jax_distributed_system: False
4-
attention: 'flash'
4+
attention: 'dot_product'
55
attention_sharding_uniform: True
66
precision: 'bf16'
77
data_sharding: ['data', 'fsdp', 'context', 'tensor']

0 commit comments

Comments
 (0)