Skip to content

Commit 2e27ca3

Browse files
committed
clean code
Signed-off-by: Yi Liu <[email protected]>
1 parent f899cbf commit 2e27ca3

File tree

2 files changed

+1
-13
lines changed

2 files changed

+1
-13
lines changed

auto_round/autoround.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1075,7 +1075,6 @@ def quant_block(self, block, input_ids, input_others, q_input=None, device=torch
10751075
total_loss = 0
10761076

10771077
for i in range(self.iters):
1078-
logger.warning(f"iter {i}")
10791078
total_loss = 0
10801079
if self.sampler == "rand":
10811080
whole_indices = torch.randperm(nsamples)[:pick_samples]

auto_round/config.py

Lines changed: 1 addition & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -13,15 +13,4 @@ class GlobalConfig:
1313
# The default values are input_backoff=0.25 and weight_backoff=0.5
1414
global_config = GlobalConfig()
1515
global_config.FP8_INPUT_BACKOFF = float(os.environ.get("AR_FP8_INPUT_BACKOFF", 1.0))
16-
global_config.FP8_WEIGHT_BACKOFF = float(os.environ.get("AR_FP8_WEIGHT_BACKOFF", 1.0))
17-
18-
19-
# AR_FP8_INPUT_BACKOFF=0.5 AR_FP8_WEIGHT_BACKOFF=1
20-
import logging
21-
logger = logging.getLogger(__name__)
22-
23-
logger.warning(f"Global config: {global_config}")
24-
25-
inc_default_config = GlobalConfig(FP8_INPUT_BACKOFF=0.25, FP8_WEIGHT_BACKOFF=0.5)
26-
config4_in_result_table = inc_default_config
27-
config5_in_result_table = GlobalConfig(FP8_INPUT_BACKOFF=0.25, FP8_WEIGHT_BACKOFF=1.0)
16+
global_config.FP8_WEIGHT_BACKOFF = float(os.environ.get("AR_FP8_WEIGHT_BACKOFF", 1.0))

0 commit comments

Comments
 (0)