From 2ab0707acbf3a3ca9e5bc5959c7920c22eba2257 Mon Sep 17 00:00:00 2001
From: Don Mahurin <dmahurin@users.noreply.github.com>
Date: Fri, 17 Nov 2023 07:32:34 -0800
Subject: [PATCH] convert : use 'model' value if it exists. This allows
 karpathy/tinyllamas to load (#4089)

Co-authored-by: Don Mahurin <@>
---
 convert.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/convert.py b/convert.py
index 3d6216f1d..5b6344aa8 100755
--- a/convert.py
+++ b/convert.py
@@ -690,6 +690,7 @@ def lazy_load_torch_file(outer_fp: IO[bytes], path: Path) -> ModelPlus:
                               data_base_path=pickle_paths[0][:-4],
                               zip_file=zf)
     model = unpickler.load()
+    if 'model' in model: model = model['model']
     as_dict = dict(model.items())
     return ModelPlus(model=as_dict, paths=[path], format='torch', vocab=None)