From 2fa3e412ce02da16f3506db8395ae35b79c13714 Mon Sep 17 00:00:00 2001
From: Don Mahurin <@>
Date: Wed, 15 Nov 2023 06:06:08 -0800
Subject: [PATCH] use 'model' value if it exists. This allows
 karpathy/tinyllamas to load.

---
 convert.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/convert.py b/convert.py
index 3d6216f1d4e7a..5b6344aa8fc85 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)