CLIP/clip
In-Ho Yi 7ef63f265b
Patch clip model for ONNX compatibility (#219)
* Patch clip model for ONNX compatibility

Changes to use INT32 for tokenization, since ONNX doesn't yet support ArgMax(INT64)
Use explicit dimension for norm

* Add compatibility fix for torch 1.7
2022-04-10 13:35:32 -07:00
..
__init__.py Make the repo installable as a package (#26) 2021-01-30 03:05:01 +09:00
bpe_simple_vocab_16e6.txt.gz Make the repo installable as a package (#26) 2021-01-30 03:05:01 +09:00
clip.py Patch clip model for ONNX compatibility (#219) 2022-04-10 13:35:32 -07:00
model.py Patch clip model for ONNX compatibility (#219) 2022-04-10 13:35:32 -07:00
simple_tokenizer.py Make the repo installable as a package (#26) 2021-01-30 03:05:01 +09:00