fix ipadapter-plus (#2065)
parent
1976cb2837
commit
f77c5eb5fa
|
|
@ -4,7 +4,7 @@ import torch
|
|||
from modules import devices
|
||||
from modules.modelloader import load_file_from_url
|
||||
from annotator.annotator_path import models_path
|
||||
from transformers import CLIPVisionModelWithProjection, CLIPVisionConfig, CLIPImageProcessor, modeling_utils
|
||||
from transformers import CLIPVisionModelWithProjection, CLIPVisionConfig, CLIPImageProcessor
|
||||
|
||||
|
||||
config_clip_g = {
|
||||
|
|
@ -77,6 +77,10 @@ downloads = {
|
|||
}
|
||||
|
||||
|
||||
clip_vision_h_uc = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'clip_vision_h_uc.data')
|
||||
clip_vision_h_uc = torch.load(clip_vision_h_uc)['uc']
|
||||
|
||||
|
||||
class ClipVisionDetector:
|
||||
def __init__(self, config):
|
||||
assert config in downloads
|
||||
|
|
|
|||
Binary file not shown.
|
|
@ -200,8 +200,9 @@ class IPAdapterModel(torch.nn.Module):
|
|||
self.image_proj_model.cpu()
|
||||
|
||||
if self.is_plus:
|
||||
from annotator.clipvision import clip_vision_h_uc
|
||||
cond = self.image_proj_model(clip_vision_output['hidden_states'][-2].to(device='cpu', dtype=torch.float32))
|
||||
uncond = self.image_proj_model(torch.zeros_like(clip_vision_output['hidden_states'][-2].to(device='cpu', dtype=torch.float32)))
|
||||
uncond = self.image_proj_model(clip_vision_h_uc.to(cond))
|
||||
return cond, uncond
|
||||
|
||||
clip_image_embeds = clip_vision_output['image_embeds'].to(device='cpu', dtype=torch.float32)
|
||||
|
|
|
|||
|
|
@ -1,4 +1,4 @@
|
|||
version_flag = 'v1.1.406'
|
||||
version_flag = 'v1.1.407'
|
||||
|
||||
from scripts.logging import logger
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue