diff --git a/src/lightning/fabric/utilities/cloud_io.py b/src/lightning/fabric/utilities/cloud_io.py index ecabbbdb6bb74..c9545265ab91d 100644 --- a/src/lightning/fabric/utilities/cloud_io.py +++ b/src/lightning/fabric/utilities/cloud_io.py @@ -13,7 +13,6 @@ # limitations under the License. """Utilities related to data saving/loading.""" -import io from pathlib import Path from typing import Any, Dict, IO, Union @@ -66,7 +65,5 @@ def _atomic_save(checkpoint: Dict[str, Any], filepath: Union[str, Path]) -> None filepath: The path to which the checkpoint will be saved. This points to the file that the checkpoint will be stored in. """ - bytesbuffer = io.BytesIO() - torch.save(checkpoint, bytesbuffer) with fsspec.open(filepath, "wb") as f: - f.write(bytesbuffer.getvalue()) + torch.save(checkpoint, f)