[loader] enable private loader (#14620)
This commit is contained in:
@@ -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):
|
||||
|
||||
Reference in New Issue
Block a user