Skip to content

Commit

Permalink
[TinyCLIP] Integrate TinyCLIP into Hugging Face Model Hub (#213)
Browse files Browse the repository at this point in the history
* [TinyCLIP] Integrate TinyCLIP into Hugging Face Model Hub
  • Loading branch information
wkcn authored Dec 20, 2023
1 parent a9e4359 commit 9002646
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions TinyCLIP/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,7 @@
* TinyCLIP ResNet-19M reduces the parameters by **50\%** while getting **$2\times$** inference speedup, and obtains **56.4\%** accuracy on ImageNet.

## News
* *Dec.2023* TinyCLIP models have been integrated into [🤗Hugging Face Model Hub](https://huggingface.co/collections/wkcn/tinyclip-model-zoo-6581aa105311fe07be88cb0d).
* *Oct.2023* Training code is released.
* *Sep.2023* This is preliminary released code, including inference code and checkpoints.

Expand All @@ -38,7 +39,7 @@ TinyCLIP ViT-45M/32 Text-18M | auto | LAION+YFCC-400M | 62.7 | 1.9 | 3,685 | [M

Note: The configs of models with auto inheritance are generated automatically.

## Getting Started
## Getting started
:beginner: Here is the setup tutorial, evaluation and pretraining scripts.

### Install dependencies and prepare dataset
Expand All @@ -47,8 +48,9 @@ Note: The configs of models with auto inheritance are generated automatically.
### Evaluate it
- [Evaluation](./docs/EVALUATION.md)

### An example for inference
### Model inference
- [Inference](./inference.py)
- [Use with 🤗Hugging Face Transformers](https://huggingface.co/collections/wkcn/tinyclip-model-zoo-6581aa105311fe07be88cb0d)

### Pretrain it
- [Pretraining](./docs/PRETRAINING.md)
Expand Down

0 comments on commit 9002646

Please sign in to comment.