File tree Expand file tree Collapse file tree 3 files changed +1
-6
lines changed
Expand file tree Collapse file tree 3 files changed +1
-6
lines changed Original file line number Diff line number Diff 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." ,
Original file line number Diff line number Diff 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 :
Original file line number Diff line number Diff 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" ,
You can’t perform that action at this time.
0 commit comments