Skip to content

Commit

Permalink
fix time was not imported
Browse files Browse the repository at this point in the history
  • Loading branch information
CSY-ModelCloud committed Feb 26, 2025
1 parent dfd10f6 commit cbb0931
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 6 deletions.
6 changes: 1 addition & 5 deletions gptqmodel/models/loader.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,16 +17,12 @@
from __future__ import annotations

import os
import time
from importlib.metadata import PackageNotFoundError, version
from typing import Dict, List, Optional, Union

import torch
import transformers

from ..nn_modules.qlinear.exllama_eora import ExllamaEoraQuantLinear
from ..nn_modules.qlinear.marlin import MarlinQuantLinear

if os.getenv('GPTQMODEL_USE_MODELSCOPE', 'False').lower() in ['true', '1']:
try:
from modelscope import snapshot_download
Expand All @@ -46,7 +42,7 @@
from ..nn_modules.qlinear.exllamav2 import ExllamaV2QuantLinear
from ..nn_modules.qlinear.ipex import IPEXQuantLinear
from ..quantization import QuantizeConfig
from ..quantization.config import FORMAT, FORMAT_FIELD_JSON, MIN_VERSION_WITH_V2
from ..quantization.config import FORMAT, MIN_VERSION_WITH_V2
from ..utils.backend import BACKEND
from ..utils.importer import auto_select_device, normalize_device_device_map, select_quant_linear
from ..utils.logger import setup_logger
Expand Down
3 changes: 2 additions & 1 deletion gptqmodel/utils/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,7 @@
import os
import re
import shutil
import time
from concurrent.futures import ThreadPoolExecutor
from enum import Enum
from typing import Any, Dict, List, Optional, Tuple, Type
Expand All @@ -50,7 +51,7 @@
from ..nn_modules.qlinear.exllamav2 import ExllamaV2QuantLinear
from ..nn_modules.qlinear.ipex import IPEXQuantLinear
from ..quantization import FORMAT, QuantizeConfig
from ..quantization.config import dynamic_get
from ..quantization.config import FORMAT_FIELD_JSON, dynamic_get
from .backend import BACKEND
from .importer import select_quant_linear
from .logger import setup_logger
Expand Down

0 comments on commit cbb0931

Please sign in to comment.