From f8cf7839356aa2b7f17567fafb828c1f5f57af59 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Tr=E1=BA=A7n=20=C4=90=E1=BB=A9c=20Nam?= Date: Tue, 19 Dec 2023 14:45:50 +0700 Subject: [PATCH] update: change order import --- convert-hf-to-gguf.py | 4 +--- convert.py | 4 +--- 2 files changed, 2 insertions(+), 6 deletions(-) diff --git a/convert-hf-to-gguf.py b/convert-hf-to-gguf.py index d1f35e0ef..52ac6fcba 100755 --- a/convert-hf-to-gguf.py +++ b/convert-hf-to-gguf.py @@ -12,8 +12,6 @@ from enum import IntEnum from pathlib import Path from typing import TYPE_CHECKING, Any, ContextManager, Iterator, cast, Optional -from awqpy.apply_awq import add_scale_weights - import numpy as np import torch @@ -999,7 +997,7 @@ def parse_args() -> argparse.Namespace: args = parse_args() if args.awq_path: - from awqpy import add_scale_weights + from awqpy.apply_awq import add_scale_weights tmp_model_path = args.model / "weighted_model" if tmp_model_path.is_dir(): print(f"{tmp_model_path} exists as a weighted model.") diff --git a/convert.py b/convert.py index 8bec7f053..3142aa8cd 100755 --- a/convert.py +++ b/convert.py @@ -23,8 +23,6 @@ from dataclasses import dataclass from pathlib import Path from typing import IO, TYPE_CHECKING, Any, Callable, Iterable, Literal, TypeVar -from awqpy.apply_awq import add_scale_weights - import numpy as np from sentencepiece import SentencePieceProcessor @@ -1156,7 +1154,7 @@ def main(args_in: list[str] | None = None) -> None: args = parser.parse_args(args_in) if args.awq_path: - from awqpy import add_scale_weights + from awqpy.apply_awq import add_scale_weights tmp_model_path = args.model / "weighted_model" if tmp_model_path.is_dir(): print(f"{tmp_model_path} exists as a weighted model.")