fix

huggingface · Oct 31, 2023 · 0e17883 · 0e17883
1 parent 7a90d64
commit 0e17883
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/optimum/intel/neural_compressor/quantization.py b/optimum/intel/neural_compressor/quantization.py
@@ -20,6 +20,7 @@
 from pathlib import Path
 from typing import Callable, Dict, Optional, Union
 
+import warnings
 import torch
 from datasets import Dataset, load_dataset
 from neural_compressor.adaptor.pytorch import PyTorch_FXAdaptor, _cfg_to_qconfig, _propagate_qconfig

diff --git a/optimum/intel/openvino/modeling_base.py b/optimum/intel/openvino/modeling_base.py
@@ -297,7 +297,7 @@ def _from_transformers(
     @classmethod
     def _to_load(
         cls,
-        model: "PreTrainedModel",
+        model,
         config: PretrainedConfig,
         onnx_config: OnnxConfig,
         use_auth_token: Optional[Union[bool, str]] = None,