From 5278174a62fa08bedcac04da605abcbeb47044cf Mon Sep 17 00:00:00 2001
From: somebody <onesomegit@outlook.com>
Date: Wed, 21 Jun 2023 17:40:47 -0500
Subject: [PATCH] Materialize on cpu

---
 modeling/patches.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/modeling/patches.py b/modeling/patches.py
index 83d157b4..dff10b93 100644
--- a/modeling/patches.py
+++ b/modeling/patches.py
@@ -205,7 +205,7 @@ def patch_transformers_for_lazyload() -> None:
 
             if isinstance(param, LazyTensor):
                 # Should always be true
-                param = param.materialize()
+                param = param.materialize(map_location="cpu")
             utils.bar.update(1)
 # END PATCH