Skip to content

Commit d590081

Browse files
authored
[Bug-Fix :] QEFFAutoModelForCausalLM __repr__() Method Fixed (#307)
This is just small fixes done for printing the `QEFFAutoModelForCausalLM`'s instance by changing the `__repr__(self)` method. Signed-off-by: Abukhoyer Shaik <[email protected]>
1 parent 01dffb6 commit d590081

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

QEfficient/transformers/models/modeling_auto.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1325,7 +1325,7 @@ def model_name(self) -> str:
13251325
return mname
13261326

13271327
def __repr__(self) -> str:
1328-
return self.__class__.__name__ + "\n" + self.model.__repr__
1328+
return self.__class__.__name__ + "\n" + self.model.__repr__()
13291329

13301330
@classmethod
13311331
@with_replaced_quantizers

0 commit comments

Comments
 (0)