Skip to content
Merged
Changes from 2 commits
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
6 changes: 2 additions & 4 deletions src/llmcompressor/modifiers/logarithmic_equalization/base.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,3 @@
from typing import List

import torch
from torch.nn import Module

Expand Down Expand Up @@ -52,8 +50,8 @@ class LogarithmicEqualizationModifier(SmoothQuantModifier):
"""

def _calculate_smoothing_scales(
self, balance_layers: List[Module], activation_scales: torch.Tensor
) -> List[float]:
self, balance_layers: list[Module], activation_scales: torch.Tensor
) -> torch.Tensor:
"""
Calculate how much smoothing to apply to each channel based on the dynamic
range of the activations and the following weights.
Expand Down