ArtFlow / artflow_model.py

Commit History

v2: Real Mamba SSM backbone (pure PyTorch), fixes torch._utils error
3239df1
verified

krystv commited on

v1.4: replace SSM scan with parallel causal linear attn, batch wavelet subbands, use F.scaled_dot_product_attention, fix AMP on CPU
0ff7ac8
verified

krystv commited on

v1.3: vectorized SSM scan, fix NaN, all 5 stage cells — artflow_model.py
0685b77
verified

krystv commited on

Add validated PyTorch prototype implementation
1866b7f
verified

krystv commited on