Skip to content

Commit 4202541

Browse files
fix time was not imported (#1349)
1 parent dfd10f6 commit 4202541

File tree

2 files changed

+3
-6
lines changed

2 files changed

+3
-6
lines changed

gptqmodel/models/loader.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -17,16 +17,12 @@
1717
from __future__ import annotations
1818

1919
import os
20-
import time
2120
from importlib.metadata import PackageNotFoundError, version
2221
from typing import Dict, List, Optional, Union
2322

2423
import torch
2524
import transformers
2625

27-
from ..nn_modules.qlinear.exllama_eora import ExllamaEoraQuantLinear
28-
from ..nn_modules.qlinear.marlin import MarlinQuantLinear
29-
3026
if os.getenv('GPTQMODEL_USE_MODELSCOPE', 'False').lower() in ['true', '1']:
3127
try:
3228
from modelscope import snapshot_download
@@ -46,7 +42,7 @@
4642
from ..nn_modules.qlinear.exllamav2 import ExllamaV2QuantLinear
4743
from ..nn_modules.qlinear.ipex import IPEXQuantLinear
4844
from ..quantization import QuantizeConfig
49-
from ..quantization.config import FORMAT, FORMAT_FIELD_JSON, MIN_VERSION_WITH_V2
45+
from ..quantization.config import FORMAT, MIN_VERSION_WITH_V2
5046
from ..utils.backend import BACKEND
5147
from ..utils.importer import auto_select_device, normalize_device_device_map, select_quant_linear
5248
from ..utils.logger import setup_logger

gptqmodel/utils/model.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@
2424
import os
2525
import re
2626
import shutil
27+
import time
2728
from concurrent.futures import ThreadPoolExecutor
2829
from enum import Enum
2930
from typing import Any, Dict, List, Optional, Tuple, Type
@@ -50,7 +51,7 @@
5051
from ..nn_modules.qlinear.exllamav2 import ExllamaV2QuantLinear
5152
from ..nn_modules.qlinear.ipex import IPEXQuantLinear
5253
from ..quantization import FORMAT, QuantizeConfig
53-
from ..quantization.config import dynamic_get
54+
from ..quantization.config import FORMAT_FIELD_JSON, dynamic_get
5455
from .backend import BACKEND
5556
from .importer import select_quant_linear
5657
from .logger import setup_logger

0 commit comments

Comments
 (0)