[loader] enable private loader (#14620)

This commit is contained in:
Yinghai Lu
2025-12-11 11:49:46 -08:00
committed by GitHub
parent 5c961756a3
commit b05b346a13
5 changed files with 13 additions and 2 deletions

View File

@@ -953,7 +953,7 @@ class ModelRunner:
return iter
def model_load_weights(model, iter):
DefaultModelLoader.load_weights_and_postprocess(model, iter, target_device)
loader.load_weights_and_postprocess(model, iter, target_device)
return model
with set_default_torch_dtype(self.model_config.dtype):