better
This commit is contained in:
@@ -10,7 +10,6 @@ def default_weight_loader(param: nn.Parameter, loaded_weight: torch.Tensor):
|
||||
|
||||
|
||||
def load_model(model: nn.Module, path: str):
|
||||
assert os.path.isdir(path)
|
||||
packed_modules_mapping = getattr(model, "packed_modules_mapping", {})
|
||||
for file in glob(os.path.join(path, "*.safetensors")):
|
||||
with safe_open(file, "pt", "cpu") as f:
|
||||
|
||||
Reference in New Issue
Block a user