Update interrogate.py
This commit is contained in:
parent
baf929821f
commit
63811455ef
@ -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
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user