We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 67dba40 commit 5ad35e1Copy full SHA for 5ad35e1
auto_round/utils/common.py
@@ -308,5 +308,8 @@ def normalize_input(decoding_layer_inputs: list[tuple[Any]]) -> Tuple[List[torch
308
input_ids.append(cur_inp[0][0][0])
309
for key, val in cur_inp[0][1].items():
310
input_others[key] = val
311
-
+ # Force 'use_cache' to be False
312
+ if "use_cache" in input_others and input_others["use_cache"] is True:
313
+ logger.warning_once("Forcing 'use_cache' to be False during calibration.")
314
+ input_others["use_cache"] = False
315
return input_ids, input_others
0 commit comments