Julien Chaumond
|
6cdc3a7844
|
[urls to hub] Replace outdated model tags with their now-canonical pipeline types (#14617)
* Replace outdated model tags with their now-canonical pipeline types
* spam the CI till it's green
|
2021-12-06 04:35:01 -05:00 |
|
Suraj Patil
|
e92190c0f8
|
Fix Flax params dtype (#13098)
* fix inits
* fix embed dtype
* fix embed dtype
* add test to check default dtype
* quality
* add type conversion methods for flax models
* more robust casting
* cast sinusoidal positions
* update pegasus
* update albert
* update test
* make sure dtype is passed to every module
* style
* fix electra dense
* fix t5
* quality
* add more tests
* better name
* use the dtype for lm head computation
* fix albert
* style
* fix albert embed dtype
* more tests
* fix vision enc-dec
* cleanup
* fix embed dtype pegasus
* fix default param test
* doc
* update template
* fix final_logits_bias dtype
* Apply suggestions from code review
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
* fix doc
* fix doc
* add detailed docstring for dtype parameter
* remove un-necessary import
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
|
2021-11-11 14:45:20 +05:30 |
|
Suraj Patil
|
85a4bda4f4
|
bump flax version (#14343)
|
2021-11-09 22:15:22 +05:30 |
|
Eduardo Gonzalez Ponferrada
|
0a22335e66
|
[Flax/run_hybrid_clip] Fix duplicating images when captions_per_image exceeds the number of captions, enable truncation
|
2021-09-02 11:19:49 +05:30 |
|
Suraj Patil
|
f5cd27694a
|
[FlaxCLIP] allow passing params to image and text feature methods (#13099)
* allow passing params to image and text feature method
* ifx for hybrid clip as well
|
2021-08-12 18:35:01 +05:30 |
|
Patrick von Platen
|
da9754a3a0
|
[Flax] Align jax flax device name (#12987)
* [Flax] Align device name in docs
* make style
* fix import error
|
2021-08-04 16:00:09 +02:00 |
|
Eduardo Gonzalez Ponferrada
|
2dd9440d08
|
Point to the right file for hybrid CLIP (#12599)
|
2021-07-12 12:16:22 +05:30 |
|
Suraj Patil
|
d7e156bd1a
|
fix loading clip vision model (#12566)
|
2021-07-07 22:50:27 +05:30 |
|
Suraj Patil
|
f5b0c1ecf0
|
[Flax] Fix hybrid clip (#12519)
* fix saving and loading
* update readme
|
2021-07-06 11:12:47 +05:30 |
|
Suraj Patil
|
23ab0b6980
|
[examples/flax] clip style image-text training example (#12491)
* clip style example
* fix post init
* add requirements
* update readme, few small fixes
|
2021-07-05 13:26:44 +05:30 |
|