Code

Fix Flax params dtype#13098

Code

Merged

patil-suraj merged 36 commits intohuggingface:masterhuggingface/transformers:masterfrom patil-suraj:fix-flax-dtypepatil-suraj/transformers:fix-flax-dtypeCopy head branch name to clipboardNov 11, 2021

+731-262

Commits

Commits on Sep 16, 2021

• fix inits

  patil-suraj

  committed
  0a4e66d

  Copy full SHA for 0a4e66d

• fix embed dtype

  patil-suraj

  committed
  e568b02

  Copy full SHA for e568b02

• fix embed dtype

  patil-suraj

  committed
  f3fc704

  Copy full SHA for f3fc704

• add test to check default dtype

  patil-suraj

  committed
  142a665

  Copy full SHA for 142a665

• quality

  patil-suraj

  committed
  d59bab1

  Copy full SHA for d59bab1

• add type conversion methods for flax models

  patil-suraj

  committed
  1e5feea

  Copy full SHA for 1e5feea

• more robust casting

  patil-suraj

  committed
  f6f92b0

  Copy full SHA for f6f92b0

• cast sinusoidal positions

  patil-suraj

  committed
  a7e1ddc

  Copy full SHA for a7e1ddc

• update pegasus

  patil-suraj

  committed
  d3209c4

  Copy full SHA for d3209c4

• update albert

  patil-suraj

  committed
  3a92bc8

  Copy full SHA for 3a92bc8

• update test

  patil-suraj

  committed
  622f215

  Copy full SHA for 622f215

• make sure dtype is passed to every module

  patil-suraj

  committed
  2f0370d

  Copy full SHA for 2f0370d

• style

  patil-suraj

  committed
  9f62f57

  Copy full SHA for 9f62f57

• fix electra dense

  patil-suraj

  committed
  cc6e655

  Copy full SHA for cc6e655

• fix t5

  patil-suraj

  committed
  a804880

  Copy full SHA for a804880

• quality

  patil-suraj

  committed
  ae2b6d1

  Copy full SHA for ae2b6d1

• add more tests

  patil-suraj

  committed
  633c0a8

  Copy full SHA for 633c0a8

• better name

  patil-suraj

  committed
  65c2455

  Copy full SHA for 65c2455

• use the dtype for lm head computation

  patil-suraj

  committed
  9dfb565

  Copy full SHA for 9dfb565

Commits on Nov 10, 2021

• Merge branch 'master' into fix-flax-dtype

  patil-suraj

  authored
  623b47f

  Copy full SHA for 623b47f

• fix albert

  patil-suraj

  committed
  fa39008

  Copy full SHA for fa39008

• style

  patil-suraj

  committed
  6c861aa

  Copy full SHA for 6c861aa

• fix albert embed dtype

  patil-suraj

  committed
  f594571

  Copy full SHA for f594571

• more tests

  patil-suraj

  committed
  084b7cb

  Copy full SHA for 084b7cb

• fix vision enc-dec

  patil-suraj

  committed
  9384382

  Copy full SHA for 9384382

• cleanup

  patil-suraj

  committed
  4b66ae4

  Copy full SHA for 4b66ae4

• fix embed dtype pegasus

  patil-suraj

  committed
  52716d6

  Copy full SHA for 52716d6

• fix default param test

  patil-suraj

  committed
  8e0572b

  Copy full SHA for 8e0572b

• doc

  patil-suraj

  committed
  347f287

  Copy full SHA for 347f287

• update template

  patil-suraj

  committed
  d26afc0

  Copy full SHA for d26afc0

• fix final_logits_bias dtype

  patil-suraj

  committed
  2b4c001

  Copy full SHA for 2b4c001

Commits on Nov 11, 2021

• Apply suggestions from code review

  Show description for 785a3e5
  

  patil-suraj

  and

  patrickvonplaten

  authored
  785a3e5

  Copy full SHA for 785a3e5

• fix doc

  patil-suraj

  committed
  a17463b

  Copy full SHA for a17463b

• fix doc

  patil-suraj

  committed
  2a993df

  Copy full SHA for 2a993df

• add detailed docstring for dtype parameter

  patil-suraj

  committed
  6dcffb8

  Copy full SHA for 6dcffb8

• remove un-necessary import

  patil-suraj

  authored
  fe8614c

  Copy full SHA for fe8614c