Skip to content

Commit 6acc231

Browse files
working; stylefixes
Signed-off-by: Brian Dellabetta <[email protected]>
1 parent 430d43d commit 6acc231

File tree

2 files changed

+2
-3
lines changed

2 files changed

+2
-3
lines changed

src/llmcompressor/modifiers/awq/base.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
import inspect
2-
from typing import Any, Dict, List, Optional, Tuple, Union
2+
from typing import Dict, List, Optional, Tuple, Union
33

44
import torch
55
from compressed_tensors.quantization import disable_quantization
@@ -24,7 +24,6 @@
2424
from llmcompressor.modifiers.quantization.calibration import update_weight_zp_scale
2525
from llmcompressor.modifiers.quantization.quantization import QuantizationMixin
2626
from llmcompressor.modifiers.utils.hooks import HooksMixin
27-
from llmcompressor.pipelines.cache import IntermediatesCache
2827
from llmcompressor.utils.fsdp.helpers import get_fsdp_parent
2928
from llmcompressor.utils.helpers import calibration_forward_context
3029
from llmcompressor.utils.pytorch.module import (

src/llmcompressor/modifiers/awq/helpers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
from typing import Tuple, Optional
1+
from typing import Optional, Tuple
22

33
import torch
44

0 commit comments

Comments
 (0)