Skip to content

Commit 70c4608

Browse files
committed
Update classifier.py
1 parent 10f7222 commit 70c4608

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

src/adaptive_classifier/classifier.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -523,7 +523,7 @@ def _save_pretrained(
523523
save_directory: Union[str, Path],
524524
config: Optional[Dict[str, Any]] = None,
525525
include_onnx: bool = True,
526-
quantize_onnx: bool = False,
526+
quantize_onnx: bool = True,
527527
**kwargs
528528
) -> Tuple[Dict[str, Any], Dict[str, Any]]:
529529
"""Save the model to a directory.
@@ -1008,7 +1008,7 @@ def push_to_hub(
10081008
self,
10091009
repo_id: str,
10101010
include_onnx: bool = True,
1011-
quantize_onnx: bool = False,
1011+
quantize_onnx: bool = True,
10121012
**kwargs
10131013
):
10141014
"""Push model to HuggingFace Hub with ONNX export by default.
@@ -1044,7 +1044,7 @@ def push_to_hub(
10441044
)
10451045

10461046
# Keep existing save/load methods for backwards compatibility
1047-
def save(self, save_dir: str, include_onnx: bool = True, quantize_onnx: bool = False):
1047+
def save(self, save_dir: str, include_onnx: bool = True, quantize_onnx: bool = True):
10481048
"""Legacy save method for backwards compatibility.
10491049
10501050
Args:

0 commit comments

Comments
 (0)