diff --git a/optimum/exporters/ggml/base.py b/optimum/exporters/ggml/base.py index 0764a527be..51688af145 100644 --- a/optimum/exporters/ggml/base.py +++ b/optimum/exporters/ggml/base.py @@ -15,7 +15,7 @@ """ggml configuration base classes.""" from abc import ABC, abstractmethod -from typing import Dict, Union +from typing import Dict, Tuple, Union from numpy import ndarray from torch import Tensor @@ -49,7 +49,7 @@ def reshape_weights(self, name: str, weights: Union[ndarray, Tensor], hparams: D @staticmethod @abstractmethod - def convert_dtype(name: str, data: Union[ndarray, Tensor], ftype: int, n_dims: int) -> tuple[ndarray, int]: + def convert_dtype(name: str, data: Union[ndarray, Tensor], ftype: int, n_dims: int) -> Tuple[ndarray, int]: return data, ftype