mirror of
https://github.com/huggingface/transformers.git
synced 2025-07-05 13:50:13 +06:00
86119c1154
* begin script * update script * fix features and data args * main * add requirements * add column name args * fix captions * don't jit transforms * fix caption * fix labels, handle attention mask * convert pixel values to numpy * labels => input_ids * transform images on the fly * use AutoModel class, create the hybird model outside of the script * fix version message * add readme * Apply suggestions from code review Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com> * adderss review comments * add more comments * allow freezing vision and text models Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
3 lines
47 B
Plaintext
3 lines
47 B
Plaintext
torch>=1.5.0
|
|
torchvision>=0.6.0
|
|
datasets>=1.8.0 |