Update README.md
Browse files
README.md
CHANGED
@@ -8,6 +8,7 @@ tags:
|
|
8 |
- pytorch_model_hub_mixin
|
9 |
- model_hub_mixin
|
10 |
pipeline_tag: zero-shot-image-classification
|
|
|
11 |
---
|
12 |
|
13 |
## Official implementation of fine-tuned ViT-H/14 ProLIP on DataComp 1B
|
@@ -68,11 +69,20 @@ print("Image uncertainty: ", i_unc)
|
|
68 |
print("Text uncertainty: ", t_unc)
|
69 |
```
|
70 |
|
71 |
-
|
72 |
-
|
73 |
-
|
74 |
-
|
75 |
-
|
76 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
77 |
}
|
78 |
```
|
|
|
8 |
- pytorch_model_hub_mixin
|
9 |
- model_hub_mixin
|
10 |
pipeline_tag: zero-shot-image-classification
|
11 |
+
library_name: prolip
|
12 |
---
|
13 |
|
14 |
## Official implementation of fine-tuned ViT-H/14 ProLIP on DataComp 1B
|
|
|
69 |
print("Text uncertainty: ", t_unc)
|
70 |
```
|
71 |
|
72 |
+
## Citation
|
73 |
+
|
74 |
+
```bibtex
|
75 |
+
@inproceedings{chun2025prolip,
|
76 |
+
title={Probabilistic Language-Image Pre-Training},
|
77 |
+
author={Chun, Sanghyuk and Kim, Wonjae and Park, Song and Yun, Sangdoo},
|
78 |
+
year={2025},
|
79 |
+
booktitle={International Conference on Learning Representations (ICLR)},
|
80 |
+
}
|
81 |
+
|
82 |
+
@inproceedings{chun2025longprolip,
|
83 |
+
title={LongProLIP: A Probabilistic Vision-Language Model with Long Context Text},
|
84 |
+
author={Chun, Sanghyuk and Yun, Sangdoo},
|
85 |
+
year={2025},
|
86 |
+
booktitle={ICLR Workshop on Quantify Uncertainty and Hallucination in Foundation Models},
|
87 |
}
|
88 |
```
|