Add library name and pipeline tag
Browse filesThis PR improves the model card by adding the `library_name` (Transformers) and the correct pipeline tag, so that the model can be found at https://huggingface.co/models?pipeline_tag=feature-extraction.
README.md
CHANGED
@@ -1,11 +1,14 @@
|
|
1 |
---
|
2 |
-
|
|
|
3 |
datasets:
|
4 |
- ILSVRC/imagenet-1k
|
5 |
- mlfoundations/datacomp_small
|
6 |
-
|
7 |
-
|
|
|
8 |
---
|
|
|
9 |
[[Paper]](https://www.arxiv.org/abs/2506.03355) [[Code]](https://github.com/LIONS-EPFL/LEAF)
|
10 |
|
11 |
Model Initialized from `openai/clip-vit-large-patch14`. The image encoder is finetuned with FARE at $\epsilon=2/255$. The text encoder is finetuned with LEAF at $k=1$ with $\rho=50$ and semantic constraints.
|
@@ -20,5 +23,4 @@ processor_name = "openai/clip-vit-large-patch14"
|
|
20 |
|
21 |
model = CLIPModel.from_pretrained(model_name)
|
22 |
processor = CLIPProcessor.from_pretrained(processor_name)
|
23 |
-
```
|
24 |
-
|
|
|
1 |
---
|
2 |
+
base_model:
|
3 |
+
- openai/clip-vit-large-patch14
|
4 |
datasets:
|
5 |
- ILSVRC/imagenet-1k
|
6 |
- mlfoundations/datacomp_small
|
7 |
+
license: mit
|
8 |
+
library_name: transformers
|
9 |
+
pipeline_tag: feature-extraction
|
10 |
---
|
11 |
+
|
12 |
[[Paper]](https://www.arxiv.org/abs/2506.03355) [[Code]](https://github.com/LIONS-EPFL/LEAF)
|
13 |
|
14 |
Model Initialized from `openai/clip-vit-large-patch14`. The image encoder is finetuned with FARE at $\epsilon=2/255$. The text encoder is finetuned with LEAF at $k=1$ with $\rho=50$ and semantic constraints.
|
|
|
23 |
|
24 |
model = CLIPModel.from_pretrained(model_name)
|
25 |
processor = CLIPProcessor.from_pretrained(processor_name)
|
26 |
+
```
|
|