ford442 commited on
Commit
5c38014
·
verified ·
1 Parent(s): 9a105d4

Update ip_adapter/ip_adapter.py

Browse files
Files changed (1) hide show
  1. ip_adapter/ip_adapter.py +1 -4
ip_adapter/ip_adapter.py CHANGED
@@ -8,10 +8,7 @@ from transformers import CLIPVisionModelWithProjection, CLIPImageProcessor
8
  from PIL import Image
9
 
10
  from .utils import is_torch2_available
11
- if is_torch2_available():
12
- from .attention_processor import IPAttnProcessor2_0 as IPAttnProcessor, AttnProcessor2_0 as AttnProcessor, CNAttnProcessor2_0 as CNAttnProcessor
13
- else:
14
- from .attention_processor import IPAttnProcessor, AttnProcessor, CNAttnProcessor
15
  from .resampler import Resampler
16
 
17
 
 
8
  from PIL import Image
9
 
10
  from .utils import is_torch2_available
11
+ from .attention_processor import IPAttnProcessor2_0 as IPAttnProcessor, AttnProcessor2_0 as AttnProcessor, CNAttnProcessor2_0 as CNAttnProcessor
 
 
 
12
  from .resampler import Resampler
13
 
14