From 63811455efa3cef42ca518d6aaa430eb57ad843e Mon Sep 17 00:00:00 2001
From: lllyasviel <lyuminzhang@outlook.com>
Date: Thu, 25 Jan 2024 12:51:14 -0800
Subject: [PATCH] Update interrogate.py

---
 modules/interrogate.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/modules/interrogate.py b/modules/interrogate.py
index 84fab496..43365a83 100644
--- a/modules/interrogate.py
+++ b/modules/interrogate.py
@@ -61,6 +61,9 @@ class InterrogateModels:
         self.offload_device = model_management.text_encoder_offload_device()
         self.dtype = torch.float32
 
+        if model_management.should_use_fp16(device=self.load_device):
+            self.dtype = torch.float16
+
         self.blip_patcher = None
         self.clip_patcher = None