42 Commits

Author SHA1 Message Date
dce31b9494 changed magma-cuda101 install cmd 2021-12-15 22:28:30 +01:00
a05aeb7ef9 added build and install instructions 2021-12-15 22:13:53 +01:00
573315e83f use pkg_resources for PyTorch version comparison (#176)
use `pkg_resources` from `setuptools` to parse version strings, it is required by Pytorch >= 0.4.1 anyway
2021-11-09 01:57:26 -05:00
1a8b4b2899 Fix PyTorch version check for nightly builds (#173) 2021-11-05 01:17:26 -04:00
2867559c5f Fix PyTorch version check (#160)
* Fix PyTorch version check

* Fix suggestion

Co-authored-by: Jong Wook Kim <jongwook@openai.com>
2021-11-04 17:32:10 -04:00
c7ba4f3907 Add 1.10.0 in tests 2021-11-04 17:20:40 -04:00
e184f608c5 Full list of Kinetics700 (Fix #158) 2021-10-19 01:40:06 -04:00
04f4dc2ca1 fix prompts format 2021-09-24 00:59:23 -04:00
4496d72f25 fix eurosat templates 2021-09-23 23:21:49 -04:00
7df882c555 add prompt data 2021-09-23 23:06:39 -04:00
efe8cbbdf3 added Country211 and Rendered SST2 dataset info 2021-09-23 21:56:48 -04:00
c13005fd42 In Compose, replace lambda function with named function (#151)
This prevents the following error on Windows (when using
a multi-process DataLoader, for example):

AttributeError: Can't pickle local object '_transform.<locals>.<lambda>'
2021-09-23 21:42:20 -04:00
3b473b0e68 Reduce half of similarity muls after encoding (#140)
(cAB)^T = c B^T A^T
Saves half of the similarity products in the CLIP model.py after the visual/text encoding stages
2021-08-29 05:15:03 -04:00
539cdcbd48 launch tests on push and PR to main (#145) 2021-08-29 05:08:37 -04:00
fa56f25251 add test.yml (#139) 2021-08-12 03:28:49 -07:00
fcab8b6eb9 Update Prompt_Engineering_for_ImageNet.ipynb (#137)
* Can specify root directory when loading model

* specifying download_root instead

* Update Prompt_Engineering_for_ImageNet.ipynb

Fix bug caused by changing default to jit=False with handling the case jit=True as well

* Reduce size of diff

* Reduce size of diff #2

* Reduce size of diff #3

* updated Interacting_with_CLIP.ipynb

* update Prompt_Engineering_for_ImageNet.ipynb

Co-authored-by: kcosta42 <kcosta@student.42.fr>
Co-authored-by: Jong Wook Kim <jongwook@openai.com>
2021-08-09 01:29:38 -07:00
22fde59cbe Can specify root directory when loading model (#136)
* Can specify root directory when loading model

* specifying download_root instead

Co-authored-by: Jong Wook Kim <jongwook@openai.com>
2021-08-08 23:43:22 -07:00
ff339871f3 Use tqdm with 1024 instead of 1000 unit scale (#131) 2021-08-08 23:20:38 -07:00
8cad3a736a Update README.md 2021-07-20 04:02:06 -07:00
dff9d15305 add ViT-B/16 and RN50x16 models 2021-07-19 14:57:31 -07:00
ea41722f9f Rename VisualTransformer -> VisionTransformer (#97)
Fixes #94
2021-07-18 20:41:49 -07:00
504cf9e491 test fix 2021-07-18 20:28:30 -07:00
a2737ac264 Add truncate option to tokenize (#126)
* Add truncate_text option to tokenize

This makes it possible to run tokenize on texts that are longer than the number of tokens
that fit the context length without having to try to guess how to cut in number of 
characters beforehand

* add doc, rename to just "truncate", use eot_token

Co-authored-by: Jong Wook Kim <jongwook@openai.com>
2021-07-18 20:17:40 -07:00
db20393f4a Using non-JIT by default; compat fix with 1.8+ 2021-07-18 18:45:21 -07:00
cfcffb90e6 add YFCC100M subset information
(#50)
2021-04-11 02:29:52 -07:00
8a665a683d fixed model loading issue (#66) 2021-03-23 03:05:17 -04:00
290ac5cb15 Correctly initializing the logit scale parameter
adding numpy import
2021-03-22 22:09:57 -04:00
43c953e231 Correctly initializing the logit scale parameter
cf. #46
2021-03-22 18:07:08 -04:00
beba48f353 Update models URIs in notebook (#52) 2021-03-07 21:58:54 -05:00
fd6c1443c2 add RN101 and RN50x4; update paper URL and model card 2021-03-04 12:30:39 -05:00
4c0275784d Load model from path (#41)
* Load model from path

* showing download progress in "MiB"

* clip.load() now can take a file path

Co-authored-by: Jong Wook Kim <jongwook@openai.com>
2021-02-16 06:19:42 -05:00
8f6deb52a1 showing download progress in "MiB" 2021-02-15 22:00:33 -05:00
e5347713f4 add prompt engineering notebook 2021-01-30 16:54:01 +09:00
d1cf43d17e using compatible version specifier in requirements.txt 2021-01-30 06:33:56 +09:00
c42a8e3c9e added parameter initialization (fixes #15) 2021-01-30 03:26:10 +09:00
4a443c18a4 updated README.md to include pip install command 2021-01-30 03:09:59 +09:00
3bee28119e Make the repo installable as a package (#26) 2021-01-30 03:05:01 +09:00
578a1d3e2e moved the notebook to subfolder 2021-01-30 00:22:03 +09:00
6bc0bd8873 added RN50 checkpoint and non-JIT model implementation 2021-01-13 08:35:50 +09:00
ebd0e35aac correctly tokenizing SOT/EOT tokens (fixes #8) 2021-01-09 02:55:09 +09:00
c89e0c16de Update the paper URL 2021-01-07 04:38:27 +09:00
b1c4b6be58 initial commit 2021-01-06 04:06:12 +09:00