Merge changes from dev

1 files changed, 2 insertions, 2 deletions

diff --git a/modules/interrogate.py b/modules/interrogate.py
index 3045560d..35a627ca 100644
--- a/modules/interrogate.py
+++ b/modules/interrogate.py
@@ -10,7 +10,7 @@ import torch.hub
 from torchvision import transforms

 from torchvision.transforms.functional import InterpolationMode

 

-from modules import devices, paths, shared, lowvram, modelloader, errors

+from modules import devices, paths, shared, lowvram, modelloader, errors, torch_utils

 

 blip_image_eval_size = 384

 clip_model_name = 'ViT-L/14'

@@ -131,7 +131,7 @@ class InterrogateModels:
 

         self.clip_model = self.clip_model.to(devices.device_interrogate)

 

-        self.dtype = next(self.clip_model.parameters()).dtype

+        self.dtype = torch_utils.get_param(self.clip_model).dtype

 

     def send_clip_to_ram(self):

         if not shared.opts.interrogate_keep_models_in_memory: