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