Skip to content

Commit 5eef9a8

Browse files
Yiozolmfracape
authored andcommitted
chore(layers): drop unused infer_swatten_* helpers
The three `infer_swatten_*` state-dict introspection helpers in `compressai/layers/attn/inference.py` were intended for MambaIC / MambaVC `from_state_dict`, but neither model is part of this PR. No code in this PR (stf.py, swin.py, tests) calls them. Removing per "don't add infrastructure for hypothetical future PRs" — the helpers can be reintroduced alongside the first model that actually uses them.
1 parent db0350c commit 5eef9a8

2 files changed

Lines changed: 0 additions & 62 deletions

File tree

compressai/layers/attn/__init__.py

Lines changed: 0 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,3 @@
1-
from .inference import (
2-
infer_swatten_attention_dim,
3-
infer_swatten_head_dim,
4-
infer_swatten_window_size,
5-
)
61
from .swin import (
72
WMSA,
83
ConvTransBlock,
@@ -30,9 +25,6 @@
3025
"WinResidualUnit",
3126
"WindowAttention",
3227
"build_window_attention_mask",
33-
"infer_swatten_attention_dim",
34-
"infer_swatten_head_dim",
35-
"infer_swatten_window_size",
3628
"pad_to_window_multiple",
3729
"window_partition",
3830
"window_reverse",

compressai/layers/attn/inference.py

Lines changed: 0 additions & 54 deletions
This file was deleted.

0 commit comments

Comments
 (0)