Skip to content

Commit

Permalink
Fixed pre-ci test error
Browse files Browse the repository at this point in the history
Signed-off-by: Cheng, Penghui <[email protected]>
  • Loading branch information
PenghuiCheng committed Mar 13, 2024
1 parent e804df3 commit 82c27dd
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions optimum/intel/neural_compressor/quantization.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@
from enum import Enum
from itertools import chain
from pathlib import Path
from typing import Callable, Dict, Optional, TypeAlias, Union
from typing import Callable, Dict, Optional, Union

import torch
from datasets import Dataset, load_dataset
Expand Down Expand Up @@ -81,9 +81,9 @@
from intel_extension_for_transformers.llm.quantization.utils import convert_to_quantized_model
from intel_extension_for_transformers.transformers.utils.config import WeightOnlyQuantConfig

Config: TypeAlias = Union[PostTrainingQuantConfig, WeightOnlyQuantConfig]
Config = Union[PostTrainingQuantConfig, WeightOnlyQuantConfig]
else:
Config: TypeAlias = PostTrainingQuantConfig
Config = PostTrainingQuantConfig

logger = logging.getLogger(__name__)

Expand Down

0 comments on commit 82c27dd

Please sign in to comment.