Skip to content

Commit

Permalink
update hessian config
Browse files Browse the repository at this point in the history
  • Loading branch information
irenaby committed Nov 12, 2024
1 parent 6a1d924 commit 269f70a
Showing 1 changed file with 1 addition and 3 deletions.
4 changes: 1 addition & 3 deletions model_compression_toolkit/gptq/common/gptq_config.py
Original file line number Diff line number Diff line change
Expand Up @@ -50,16 +50,14 @@ class GPTQHessianScoresConfig:
hessians_num_samples: Optional[int]
norm_scores: bool = None
log_norm: bool = None
scale_log_norm: bool = None
scale_log_norm: bool = False
hessian_batch_size: int = ACT_HESSIAN_DEFAULT_BATCH_SIZE

def __post_init__(self):
if self.norm_scores is None:
self.norm_scores = not self.per_sample
if self.log_norm is None:
self.log_norm = not self.per_sample
if self.scale_log_norm is None:
self.scale_log_norm = False


@dataclass
Expand Down

0 comments on commit 269f70a

Please sign in to comment.