From 3fe2519852b9a7f43fba33ec03c82337abc2fc6a Mon Sep 17 00:00:00 2001
From: janEbert <janpublicebert@posteo.net>
Date: Thu, 6 Mar 2025 11:04:48 +0100
Subject: [PATCH] Clean up whitespace
---
pytorch-fsdp-example/main.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/pytorch-fsdp-example/main.py b/pytorch-fsdp-example/main.py
index d6ea289..c72dd78 100644
--- a/pytorch-fsdp-example/main.py
+++ b/pytorch-fsdp-example/main.py
@@ -127,7 +127,7 @@ def save_model(model, save_dir):
options=state_dict_options,
)
cp_state_dict = {'model': model_state_dict}
-
+
dcp.save(
cp_state_dict,
storage_writer=dcp.FileSystemWriter(save_dir, overwrite=True),
@@ -155,7 +155,7 @@ def load_model(model, load_dir):
options=state_dict_options,
)
cp_state_dict = {'model': model_state_dict}
-
+
dcp.load(
cp_state_dict,
storage_reader=dcp.FileSystemReader(load_dir),
--
GitLab