diff --git a/eland/ml/pytorch/transformers.py b/eland/ml/pytorch/transformers.py index 89e4634..9b2b78e 100644 --- a/eland/ml/pytorch/transformers.py +++ b/eland/ml/pytorch/transformers.py @@ -578,7 +578,7 @@ class TransformerModel: model_id: str, task_type: str, *, - es_version: Optional[Tuple[int, int, int]], + es_version: Optional[Tuple[int, int, int]] = None, quantize: bool = False, ): """