diff options
| author | Aarni Koskela <akx@iki.fi> | 2023-12-30 22:20:30 +0000 |
|---|---|---|
| committer | Aarni Koskela <akx@iki.fi> | 2023-12-31 11:22:43 +0000 |
| commit | 5768afc776a66bb94e77a9c1daebeea58fa731d5 (patch) | |
| tree | 06c015a95076b2de579a72750aa7862a9615ed09 /modules/interrogate.py | |
| parent | a84e842189f5599fd354147f72d1a9b9ed0716c8 (diff) | |
| download | stable-diffusion-webui-gfx803-5768afc776a66bb94e77a9c1daebeea58fa731d5.tar.gz stable-diffusion-webui-gfx803-5768afc776a66bb94e77a9c1daebeea58fa731d5.tar.bz2 stable-diffusion-webui-gfx803-5768afc776a66bb94e77a9c1daebeea58fa731d5.zip | |
Add utility to inspect a model's parameters (to get dtype/device)
Diffstat (limited to 'modules/interrogate.py')
| -rw-r--r-- | modules/interrogate.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/modules/interrogate.py b/modules/interrogate.py index 3045560d..5be5a10f 100644 --- a/modules/interrogate.py +++ b/modules/interrogate.py @@ -11,6 +11,7 @@ from torchvision import transforms from torchvision.transforms.functional import InterpolationMode
from modules import devices, paths, shared, lowvram, modelloader, errors
+from modules.torch_utils import get_param
blip_image_eval_size = 384
clip_model_name = 'ViT-L/14'
@@ -131,7 +132,7 @@ class InterrogateModels: self.clip_model = self.clip_model.to(devices.device_interrogate)
- self.dtype = next(self.clip_model.parameters()).dtype
+ self.dtype = get_param(self.clip_model).dtype
def send_clip_to_ram(self):
if not shared.opts.interrogate_keep_models_in_memory:
|