Update convert_to_singleton.py

Delete temp file saving

Update convert_to_singleton.py
Delete temp file saving
fd353a6f · Arthur · GitHub · 2f6bec11 · fd353a6f
Unverified Commit fd353a6f authored 3 years ago by Arthur Committed by GitHub 3 years ago
Hide whitespace changes
Inline Side-by-side

Showing

with 0 additions and 4 deletions
+0 -4
--- a/metaseq/scripts/convert_to_singleton.py
+++ b/metaseq/scripts/convert_to_singleton.py
@@ -93,10 +93,6 @@ def worker_main(cfg: MetaseqConfig):

    if dist_utils.get_global_rank() == 0:
        glued = glue_megatron_parts(model_parts)
-
-        with open("temp.pt", "wb") as f:
-            torch.save(glued, f)
-
        glued["decoder.version"] = torch.tensor([3])

        if "decoder.output_projection.weight" in glued: