From c56bc2994dbb98ebdb08201745d3f2d220dab14a Mon Sep 17 00:00:00 2001
From: CarlOwOs <carloshurtadocomin@gmail.com>
Date: Thu, 1 Jan 2026 12:57:37 +0000
Subject: [PATCH] fixed hf to dcp conversion by removing the double "model."
 prefix

---
 flame/utils/convert_hf_to_dcp.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/flame/utils/convert_hf_to_dcp.py b/flame/utils/convert_hf_to_dcp.py
index bab94eb..5ceb001 100644
--- a/flame/utils/convert_hf_to_dcp.py
+++ b/flame/utils/convert_hf_to_dcp.py
@@ -21,7 +21,7 @@ def convert_hf_weights(model: str, checkpoint: str):
     logger.info(f"Writing to DCP at '{checkpoint}'")
     checkpoint.mkdir(parents=True, exist_ok=True)
     storage_writer = DCP.filesystem.FileSystemWriter(checkpoint, thread_count=8)
-    DCP.save({"model": state_dict}, storage_writer=storage_writer)
+    DCP.save(state_dict, storage_writer=storage_writer)
 
 
 if __name__ == "__main__":