Skip to content

Commit 2cf2aa9

Browse files
committed
change log level
Signed-off-by: Kyle Sayers <kylesayrs@gmail.com>
1 parent 3ba4f00 commit 2cf2aa9

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/llmcompressor/modeling/moe_context.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -118,7 +118,7 @@ def moe_calibration_context(
118118
replaced = {}
119119

120120
# Step 1: Collect all MoE modules that need replacement
121-
logger.info("Entering MoE calibration context")
121+
logger.debug("Entering MoE calibration context")
122122
modules_to_replace = []
123123
for name, module in model.named_modules():
124124
class_name = module.__class__.__name__
@@ -161,7 +161,7 @@ def moe_calibration_context(
161161
yield
162162
finally:
163163
# Step 2: Restore non-permanent modules
164-
for name, (original, replacement) in replaced.items():
164+
for name, (_original, replacement) in replaced.items():
165165
if not replacement.is_permanent:
166166
restored = replacement.restore(original)
167167
model.set_submodule(name, restored)

0 commit comments

Comments
 (0)