NielsRogge/transformers

Pull Requests Commits

Address @bastings comment in PR, moving jax.jit decorated outside of __call__

mfuntowicz committed 5 years ago

49303a2d

Fix invalid number of arguments when raising issue for an unknown model.

mfuntowicz committed 5 years ago

c2b84b0e

Simplify Flax unittests to avoid machine crashes.

mfuntowicz committed 5 years ago

c43ee15e

mfuntowicz committed 5 years ago

286de82f

Fix too much parameters in BertAttention.

mfuntowicz committed 5 years ago

bf8f09b8

trigger ci again

mfuntowicz committed 5 years ago

1a8228b4

Put seed as optional constructor parameter.

mfuntowicz committed 5 years ago

bc7b8720

Remove unused import.

mfuntowicz committed 5 years ago

5435a826

Change the type of the catch for msgpack.

mfuntowicz committed 5 years ago

b872af5f

Make flake8 happy, remove unused jax import

mfuntowicz committed 5 years ago

a70c841b

Bump and pin flax release to 0.1.0.

mfuntowicz committed 5 years ago

097d77c3

Fix another invalid import in flax_roberta test

mfuntowicz committed 5 years ago

e020cad2

Fix invalid is_<framework>_available call, thanks @LysandreJik :tada:

mfuntowicz committed 5 years ago

91c8daf8

Try removing trailing comma to make both black and flake happy!

mfuntowicz committed 5 years ago

b9a6555f

When black and flake8 fight each other for a space ... :boom: :boom: :boom:

mfuntowicz committed 5 years ago

f0ad7ef6

Redo black again and again

mfuntowicz committed 5 years ago

11862199

Attempt to fix flake8 warnings ...

mfuntowicz committed 5 years ago

776b4d35

Added jax & jaxlib to the flax optional dependencies.

mfuntowicz committed 5 years ago

0e6a0560

Attempt to make style in tests too

mfuntowicz committed 5 years ago

f4e9556b

Attempt to make style

mfuntowicz committed 5 years ago

00564f89

Fix typo suggested by @stefan-it

mfuntowicz committed 5 years ago

f20a9070

Expose seed in both Bert and Roberta

mfuntowicz committed 5 years ago

7b79d659

Addressing review comments.

mfuntowicz committed 5 years ago

1cd52e61

Flax model reference the _MODEL_ARCHIVE_MAP from PyTorch model.

mfuntowicz committed 5 years ago

6a393c9b

Added more unittests

mfuntowicz committed 5 years ago

54bcaa65

Introduce FlaxAutoModel

mfuntowicz committed 5 years ago

8a87b0f1

Added run_tests_flax to the CI.

mfuntowicz committed 5 years ago

12a1c0b2

Added unittest for Flax models.

mfuntowicz committed 5 years ago

254fbbef

Expose `is_flax_available` in file_utils.

mfuntowicz committed 5 years ago

ef70e9bc

Use specialized, model dependent conversion functio.

mfuntowicz committed 5 years ago

e2807189

Older