File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
src/llmcompressor/modeling Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff 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 )
You can’t perform that action at this time.
0 commit comments