From ba1a93429756b72f2cbd4f8b166d06cbd65ab55d Mon Sep 17 00:00:00 2001
From: Brandon Rising <brandon@invoke.ai>
Date: Thu, 20 Jul 2023 14:02:23 -0400
Subject: [PATCH] Fix Lora typings

---
 invokeai/backend/model_management/lora.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/invokeai/backend/model_management/lora.py b/invokeai/backend/model_management/lora.py
index 37e5a6fa40..d78289e7e2 100644
--- a/invokeai/backend/model_management/lora.py
+++ b/invokeai/backend/model_management/lora.py
@@ -767,6 +767,7 @@ class ONNXModelPatcher:
                     if not layer_key.startswith(prefix):
                         continue
 
+                    layer.to(dtype=torch.float32)
                     layer_key = layer_key.replace(prefix, "")
                     layer_weight = layer.get_weight().detach().cpu().numpy() * lora_weight
                     if layer_key is blended_loras: