Skip to content

Commit d0fd3e0

Browse files
authored
fix: correct ZipAR tag and remove non existent tags (#573)
1 parent ab8dccf commit d0fd3e0

File tree

3 files changed

+1
-6
lines changed

3 files changed

+1
-6
lines changed

src/pruna/algorithms/base/tags.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -75,10 +75,6 @@ class AlgorithmTag(Enum):
7575
"enhancer",
7676
"Enhancers improve the quality of the model's output. Enhancers can range from post-processing to test time compute algorithms.",
7777
)
78-
RESAMPLER = (
79-
"resampler",
80-
"Resamplers change the shape of image or video latents during generation to speed up inference.",
81-
)
8278
RECOVERER = (
8379
"recoverer",
8480
"Recovery restores the performance of a model after compression.",

src/pruna/algorithms/hyper.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,6 @@ class Hyper(PrunaAlgorithmBase):
5959
"torch_compile",
6060
"stable_fast",
6161
AlgorithmTag.ENHANCER, # type: ignore[attr-defined]
62-
AlgorithmTag.RESAMPLER, # type: ignore[attr-defined]
6362
]
6463

6564
def get_hyperparameters(self) -> list:

src/pruna/algorithms/zipar/zipar.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ class ZipAR(PrunaAlgorithmBase):
3535
"""
3636

3737
algorithm_name: str = "zipar"
38-
group_tags: list[AlgorithmTag] = [AlgorithmTag.COMPILER]
38+
group_tags: list[AlgorithmTag] = [AlgorithmTag.DECODER]
3939
save_fn = SAVE_FUNCTIONS.reapply
4040
references: dict[str, str] = {
4141
"GitHub": "https://github.com/thisisbillhe/zipar",

0 commit comments

Comments
 (0)