fix: switch torch_dynamic and quanto saves to save_before_apply

src/pruna/algorithms/quanto.py

@@ -44,7 +44,7 @@ class Quanto(PrunaAlgorithmBase):
     algorithm_name: str = "quanto"
     group_tags: list[tags] = [tags.QUANTIZER]
     references: dict[str, str] = {"GitHub": "https://github.com/huggingface/optimum-quanto"}
-    save_fn: SAVE_FUNCTIONS = SAVE_FUNCTIONS.reapply
+    save_fn: SAVE_FUNCTIONS = SAVE_FUNCTIONS.save_before_apply
     tokenizer_required: bool = False
     processor_required: bool = False
     dataset_required: bool = False

src/pruna/algorithms/torch_dynamic.py

@@ -37,7 +37,7 @@ class TorchDynamic(PrunaAlgorithmBase):
     algorithm_name = "torch_dynamic"
     group_tags: list[tags] = [tags.QUANTIZER]
     references: dict[str, str] = {"GitHub": "https://github.com/pytorch/pytorch"}
-    save_fn: SAVE_FUNCTIONS = SAVE_FUNCTIONS.pickled
+    save_fn: SAVE_FUNCTIONS = SAVE_FUNCTIONS.save_before_apply
     tokenizer_required: bool = False
     processor_required: bool = False
     runs_on: list[str] = ["cpu", "cuda"]

tests/algorithms/testers/torch_dynamic.py

@@ -6,7 +6,7 @@
 class TestTorchDynamic(AlgorithmTesterBase):
     """Test the torch dynamic quantizer."""
 
-    models = ["shufflenet"]
+    models = ["shufflenet", "smollm_135m"]
     reject_models = []
     allow_pickle_files = False
     algorithm_class = TorchDynamic