Skip to content

Commit d69dfd8

Browse files
authored
HOTFIX/fixed replace quantizers (quic#273)
Signed-off-by: Onkar Chougule <quic_ochougul@quicinc.com>
1 parent b6989c0 commit d69dfd8

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

QEfficient/transformers/models/modeling_auto.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1259,6 +1259,7 @@ def __repr__(self) -> str:
12591259
return self.__class__.__name__ + "\n" + self.model.__repr__
12601260

12611261
@classmethod
1262+
@with_replaced_quantizers
12621263
def from_pretrained(
12631264
cls, pretrained_model_name_or_path, continuous_batching: bool = False, is_tlm: bool = False, *args, **kwargs
12641265
):

0 commit comments

Comments
 (0)