Skip to content

Commit e053186

Browse files
feat: decoder algorithm (#444)
* init commit * added license * fixing linting error * small tweaks * pinning transformer version and added missing parameter * added zipar extra and runtime version check for Janus compatibility * Co-author Co-authored-by: Gaspar Rochette <gaspar.rochette@pruna.ai> * added warning to about transformer version * ci: retrigger CI * adde decoder tag * remove zipar extra * fixing typo --------- Co-authored-by: Gaspar Rochette <gaspar.rochette@pruna.ai>
1 parent 223ff38 commit e053186

File tree

4 files changed

+843
-0
lines changed

4 files changed

+843
-0
lines changed

src/pruna/algorithms/base/tags.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -79,6 +79,10 @@ class AlgorithmTag(Enum):
7979
"resampler",
8080
"Resamplers change the shape of image or video latents during generation to speed up inference.",
8181
)
82+
DECODER = (
83+
"decoder",
84+
"Decoders speed up autoregressive generation by changing their decoding strategy to be more parallelizable.",
85+
)
8286

8387
def __init__(self, name: str, description: str):
8488
"""
Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,17 @@
1+
# Copyright 2025 - Pruna AI GmbH. All rights reserved.
2+
#
3+
# Licensed under the Apache License, Version 2.0 (the "License");
4+
# you may not use this file except in compliance with the License.
5+
# You may obtain a copy of the License at
6+
#
7+
# http://www.apache.org/licenses/LICENSE-2.0
8+
#
9+
# Unless required by applicable law or agreed to in writing, software
10+
# distributed under the License is distributed on an "AS IS" BASIS,
11+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12+
# See the License for the specific language governing permissions and
13+
# limitations under the License.
14+
15+
from pruna.algorithms.zipar.zipar import ZipAR
16+
17+
__all__ = ["ZipAR"]

0 commit comments

Comments
 (0)