Add model

README.md

@@ -0,0 +1,68 @@
+---
+tags:
+- clip
+library_name: open_clip
+pipeline_tag: zero-shot-image-classification
+license: apache-2.0
+datasets:
+- mlfoundations/datacomp_1b
+---
+# Model card for ViT-H-14-CLIPA-336-datacomp1B
+
+A CLIPA-v2 model...
+
+## Model Details
+- **Model Type:** Contrastive Image-Text, Zero-Shot Image Classification.
+- **Original:** https://github.com/UCSC-VLAA/CLIPA
+- **Dataset:** mlfoundations/datacomp_1b
+- **Papers:**
+  - CLIPA-v2: Scaling CLIP Training with 81.1% Zero-shot ImageNet Accuracy within a $10,000 Budget; An Extra $4,000 Unlocks 81.8% Accuracy: https://arxiv.org/abs/2306.15658
+  - An Inverse Scaling Law for CLIP Training: https://arxiv.org/abs/2305.07017
+
+## Model Usage
+### With OpenCLIP
+```
+import torch
+import torch.nn.functional as F
+from urllib.request import urlopen
+from PIL import Image
+from open_clip import create_model_from_pretrained, get_tokenizer
+
+model, preprocess = create_model_from_pretrained('hf-hub:ViT-H-14-CLIPA-336')
+tokenizer = get_tokenizer('hf-hub:ViT-H-14-CLIPA-336')
+
+image = Image.open(urlopen(
+    'https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/beignets-task-guide.png'
+))
+image = preprocess(image).unsqueeze(0)
+
+text = tokenizer(["a diagram", "a dog", "a cat", "a beignet"], context_length=model.context_length)
+
+with torch.no_grad(), torch.cuda.amp.autocast():
+    image_features = model.encode_image(image)
+    text_features = model.encode_text(text)
+    image_features = F.normalize(image_features, dim=-1)
+    text_features = F.normalize(text_features, dim=-1)
+
+    text_probs = (100.0 * image_features @ text_features.T).softmax(dim=-1)
+
+print("Label probs:", text_probs)  # prints: [[0., 0., 0., 1.0]]
+```   
+
+## Citation
+```bibtex
+@article{li2023clipav2,
+      title={CLIPA-v2: Scaling CLIP Training with 81.1% Zero-shot ImageNet Accuracy within a $10,000 Budget; An Extra $4,000 Unlocks 81.8% Accuracy}, 
+      author={Xianhang Li and Zeyu Wang and Cihang Xie},
+      journal={arXiv preprint arXiv:2306.15658},
+      year={2023},
+}
+```
+```bibtex
+@inproceedings{li2023clipa,
+      title={An Inverse Scaling Law for CLIP Training}, 
+      author={Xianhang Li and Zeyu Wang and Cihang Xie},
+      booktitle={NeurIPS},
+      year={2023},
+}
+```

added_tokens.json

@@ -0,0 +1,7 @@
+{
+  "[CLS]": 101,
+  "[MASK]": 103,
+  "[PAD]": 0,
+  "[SEP]": 102,
+  "[UNK]": 100
+}

open_clip_config.json

@@ -0,0 +1,42 @@
+{
+  "model_cfg": {
+    "embed_dim": 1024,
+    "vision_cfg": {
+      "image_size": 336,
+      "layers": 32,
+      "width": 1280,
+      "head_width": 80,
+      "patch_size": 14,
+      "no_ln_pre": true,
+      "pool_type": "avg",
+      "final_ln_after_pool": true
+    },
+    "text_cfg": {
+      "context_length": 32,
+      "vocab_size": 32000,
+      "hf_tokenizer_name": "bert-base-uncased",
+      "tokenizer_kwargs": {
+        "strip_sep_token": true
+      },
+      "width": 1024,
+      "heads": 16,
+      "layers": 24,
+      "pool_type": "last",
+      "no_causal_mask": true
+    }
+  },
+  "preprocess_cfg": {
+    "mean": [
+      0.485,
+      0.456,
+      0.406
+    ],
+    "std": [
+      0.229,
+      0.224,
+      0.225
+    ],
+    "interpolation": "bilinear",
+    "resize_mode": "squash"
+  }
+}

open_clip_model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7d3d8265422fd4791425d745705f3093a807b56d7bf0770fc51ebc7327976925
+size 3874658260

open_clip_pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:dcfc8340e841d3a0436c370466472cddeb69e2e60d9a7d1180c6f56a92397e67
+size 3874844314

special_tokens_map.json

@@ -0,0 +1,7 @@
+{
+  "cls_token": "[CLS]",
+  "mask_token": "[MASK]",
+  "pad_token": "[PAD]",
+  "sep_token": "[SEP]",
+  "unk_token": "[UNK]"
+}

tokenizer_config.json

@@ -0,0 +1,56 @@
+{
+  "added_tokens_decoder": {
+    "0": {
+      "content": "[PAD]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "100": {
+      "content": "[UNK]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "101": {
+      "content": "[CLS]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "102": {
+      "content": "[SEP]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "103": {
+      "content": "[MASK]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    }
+  },
+  "additional_special_tokens": [],
+  "clean_up_tokenization_spaces": true,
+  "cls_token": "[CLS]",
+  "do_lower_case": true,
+  "mask_token": "[MASK]",
+  "model_max_length": 512,
+  "pad_token": "[PAD]",
+  "sep_token": "[SEP]",
+  "strip_accents": null,
+  "tokenize_chinese_chars": true,
+  "tokenizer_class": "BertTokenizer",
+  "unk_token": "[UNK]"
+}