primerz commited on
Commit
66b6852
·
verified ·
1 Parent(s): 04fa627

Update models.py

Browse files
Files changed (1) hide show
  1. models.py +4 -1
models.py CHANGED
@@ -13,7 +13,10 @@ from diffusers import (
13
  LCMScheduler
14
  )
15
  from diffusers.models.attention_processor import AttnProcessor2_0
16
- from transformers import CLIPVisionModelWithProjection
 
 
 
17
  from insightface.app import FaceAnalysis
18
  from controlnet_aux import ZoeDetector, OpenposeDetector, LeresDetector, MidasDetector, MediapipeFaceDetector
19
  from huggingface_hub import hf_hub_download, snapshot_download
 
13
  LCMScheduler
14
  )
15
  from diffusers.models.attention_processor import AttnProcessor2_0
16
+ from transformers import (
17
+ CLIPVisionModelWithProjection, CLIPTokenizer,
18
+ CLIPTextModel, CLIPTextModelWithProjection
19
+ )
20
  from insightface.app import FaceAnalysis
21
  from controlnet_aux import ZoeDetector, OpenposeDetector, LeresDetector, MidasDetector, MediapipeFaceDetector
22
  from huggingface_hub import hf_hub_download, snapshot_download