Trent commited on
Commit
c0d4b59
β€’
1 Parent(s): cf349fd

Model architecture details

Browse files
Files changed (1) hide show
  1. text2image.py +2 -0
text2image.py CHANGED
@@ -23,6 +23,8 @@ def app(model_name):
23
 
24
  KoCLIP is a retraining of OpenAI's CLIP model using 82,783 images from [MSCOCO](https://cocodataset.org/#home) dataset and
25
  Korean caption annotations. Korean translation of caption annotations were obtained from [AI Hub](https://aihub.or.kr/keti_data_board/visual_intelligence).
 
 
26
 
27
  Example Queries : μ•„νŒŒνŠΈ(Apartment), μžλ™μ°¨(Car), 컴퓨터(Computer)
28
  """)
 
23
 
24
  KoCLIP is a retraining of OpenAI's CLIP model using 82,783 images from [MSCOCO](https://cocodataset.org/#home) dataset and
25
  Korean caption annotations. Korean translation of caption annotations were obtained from [AI Hub](https://aihub.or.kr/keti_data_board/visual_intelligence).
26
+ Base model `koclip` uses `klue/roberta` as text encoder and `openai/clip-vit-base-patch32` as image encoder.
27
+ Larger model `koclip-large` uses `klue/roberta` as text encoder and bigger `google/vit-large-patch16-224` as image encoder.
28
 
29
  Example Queries : μ•„νŒŒνŠΈ(Apartment), μžλ™μ°¨(Car), 컴퓨터(Computer)
30
  """)