diff --git a/library/safetensors_utils.py b/library/safetensors_utils.py index dcd2309e..c65cdfab 100644 --- a/library/safetensors_utils.py +++ b/library/safetensors_utils.py @@ -44,7 +44,6 @@ def mem_eff_save_file(tensors: Dict[str, torch.Tensor], filename: str, metadata: validated[key] = value return validated - # print(f"Using memory efficient save file: {filename}") header = {} offset = 0