Keep the fix itself (size-based chunking/capping) but drop the _diag_log_attn_size helper, the debug-level [MPS-ATTN-FIX:...] messages, and the now-unused tag parameters that only existed to support that logging -- a minimal correctness fix without added observability scaffolding. Co-Authored-By: Claude Sonnet 5 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_01HAMqHgFBD6e9wjU8k8U6uW |
||
|---|---|---|
| .. | ||
| diffusionmodules | ||
| distributions | ||
| encoders | ||
| attention.py | ||
| ema.py | ||
| sdpose.py | ||
| sub_quadratic_attention.py | ||
| temporal_ae.py | ||