hoelee/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-08-03 10:50:23 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

change import statements for #14478

Browse Source
This commit is contained in:
AUTOMATIC1111
2023-12-31 22:38:30 +03:00
parent be5f1acc8f
commit a70dfb64a8
7 changed files with 14 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
modules/interrogate.py
View File

@@ -10,8 +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.torch_utils import get_param
from modules import devices, paths, shared, lowvram, modelloader, errors, torch_utils
blip_image_eval_size = 384
clip_model_name = 'ViT-L/14'
@@ -132,7 +131,7 @@ class InterrogateModels:
self.clip_model = self.clip_model.to(devices.device_interrogate)
self.dtype = get_param(self.clip_model).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: