Delete IP-Adapter copies of AttnProcessor and AttnProcessor2_0, which were unmodified from diffusers.

This commit is contained in:
Ryan Dick
2023-09-07 14:10:42 -04:00
parent 23fdf0156f
commit 7703bf2ca1
2 changed files with 17 additions and 160 deletions

View File

@ -4,8 +4,6 @@
from typing import List
import torch
from PIL import Image
from transformers import CLIPImageProcessor, CLIPVisionModelWithProjection
# FIXME: Getting errors when trying to use PyTorch 2.0 versions of IPAttnProcessor and AttnProcessor
# so for now falling back to the default versions
@ -14,6 +12,9 @@ from transformers import CLIPImageProcessor, CLIPVisionModelWithProjection
# from .attention_processor import IPAttnProcessor2_0 as IPAttnProcessor, AttnProcessor2_0 as AttnProcessor
# else:
# from .attention_processor import IPAttnProcessor, AttnProcessor
from PIL import Image
from transformers import CLIPImageProcessor, CLIPVisionModelWithProjection
from .attention_processor import AttnProcessor, IPAttnProcessor
from .resampler import Resampler