Skip to content

Commit 9cbda36

Browse files
committed
Minor changes
Signed-off-by: Dipankar Sarkar <[email protected]>
1 parent 77b6966 commit 9cbda36

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

QEfficient/transformers/models/pytorch_transforms.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -500,7 +500,7 @@ class SpDTransform:
500500
@classmethod
501501
def apply(cls, model: nn.Module, qaic_config: Optional[dict] = None, **kwargs) -> Tuple[nn.Module, bool]:
502502
transformed = False
503-
kwargs.pop("pretrained_model_name_or_path")
503+
kwargs.pop("pretrained_model_name_or_path", None)
504504

505505
if qaic_config is None or (speculative_model_type := qaic_config.get("speculative_model_type")) is None:
506506
return model, transformed

0 commit comments

Comments
 (0)