Skip to content
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
fix style
Signed-off-by: changwangss <[email protected]>
  • Loading branch information
changwangss committed Aug 26, 2024
commit ea62bb352e5b68ef42bff8e0048c518faf743cca
5 changes: 1 addition & 4 deletions neural_compressor/transformers/models/modeling_auto.py
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,6 @@
from threading import Thread
from typing import Union

import torch
import transformers
from accelerate import init_empty_weights
from transformers import AutoConfig
Expand All @@ -56,12 +55,10 @@
from neural_compressor.torch.algorithms.weight_only.modules import INCWeightOnlyLinear
from neural_compressor.torch.utils import is_ipex_available
from neural_compressor.transformers import GPTQConfig, RtnConfig
from neural_compressor.transformers.quantization.utils import replace_linear, save_low_bit
from neural_compressor.transformers.quantization.utils import convert_dtype_torch2str, replace_linear, save_low_bit
from neural_compressor.utils import logger
from neural_compressor.utils.utility import CpuInfo, LazyImport

from ..quantization.utils import convert_dtype_torch2str, replace_linear

torch = LazyImport("torch")


Expand Down
Loading