remove hard-coded value

Cyrilvallez · Cyrilvallez · commit d8acc79f89f0 · 2025-11-18T10:01:15.000+01:00
diff --git a/src/transformers/modeling_utils.py b/src/transformers/modeling_utils.py
@@ -4320,7 +4320,7 @@ def _load_pretrained_model(
             misc=misc,
             ignore_mismatched_sizes=ignore_mismatched_sizes,
         )
-        disk_offload_index = None
+
         return model, missing_keys, unexpected_keys, mismatched_keys, disk_offload_index, error_msgs
 
     def retrieve_modules_from_names(self, names, add_prefix=False, remove_prefix=False):

Original file line number	Diff line number	Diff line change
`@@ -4320,7 +4320,7 @@ def _load_pretrained_model(`
`4320`	`4320`	`misc=misc,`
`4321`	`4321`	`ignore_mismatched_sizes=ignore_mismatched_sizes,`
`4322`	`4322`	`)`
`4323`		`- disk_offload_index = None`
	`4323`	`+`
`4324`	`4324`	`return model, missing_keys, unexpected_keys, mismatched_keys, disk_offload_index, error_msgs`
`4325`	`4325`
`4326`	`4326`	`def retrieve_modules_from_names(self, names, add_prefix=False, remove_prefix=False):`