From d0d215bb37f4eb81e0355b78315d505e52743c25 Mon Sep 17 00:00:00 2001 From: somebody Date: Wed, 31 May 2023 10:59:03 -0500 Subject: [PATCH] I have no idea why this was removed i swear it was there --- modeling/patches.py | 1 + 1 file changed, 1 insertion(+) diff --git a/modeling/patches.py b/modeling/patches.py index 2b823b3b..203d98c9 100644 --- a/modeling/patches.py +++ b/modeling/patches.py @@ -290,6 +290,7 @@ def patch_transformers_for_lazyload() -> None: tensor_name=param_name, device=param_device, value=param, + dtype=dtype, ) else: if (