lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-02-23 08:13:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update interrogate.py

Browse Source
This commit is contained in:
lllyasviel
2024-01-25 12:51:14 -08:00
parent baf929821f
commit 63811455ef
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
modules/interrogate.py
View File

@@ -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