fix ipadapter-plus (#2065)

pull/2068/head
lllyasviel 2023-09-05 21:35:10 -07:00 committed by GitHub
parent 1976cb2837
commit f77c5eb5fa
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 8 additions and 3 deletions

View File

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

View File

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

View File

@ -1,4 +1,4 @@
version_flag = 'v1.1.406'
version_flag = 'v1.1.407'
from scripts.logging import logger