Implement `FlaxModelMixin` - #493

Merged

mishig25 merged 20 commits into

mainfrom

flax_model_mixin

Sep 14, 2022

Commits on Sep 13, 2022

Implement `FlaxModelMixin`
mishig25
committed
95073e1

Commits on Sep 14, 2022

Rm unused method `framework`
mishig25
committed
91559f3
Update src/diffusers/modeling_flax_utils.py

Mishig Davaadorj
and
patil-suraj
authored
f7a0ab2
some more changes
patrickvonplaten
committed
5d81bf8
make style
patrickvonplaten
committed
1430ab8
Add comment
mishig25
committed
6a2a4c1
Update src/diffusers/modeling_flax_utils.py

Mishig Davaadorj
and
patrickvonplaten
authored
2bf0267
Rm unneeded comment
mishig25
committed
25ab3ca
Update docstrings
mishig25
committed
1e8466e
correct ignore kwargs
patrickvonplaten
committed
6842d29
Merge branch 'flax_model_mixin' of https://github.com/huggingface/diffusers into flax_model_mixin
patrickvonplaten
committed
4f6b01b
make style
patrickvonplaten
committed
0f26c05
Update docstring examples
mishig25
committed
d98e8c7
Merge branch 'flax_model_mixin' of https://github.com/huggingface/diffusers into flax_model_mixin
mishig25
committed
5a7b784
Make style
mishig25
committed
5d08577
Update src/diffusers/modeling_flax_utils.py

Mishig Davaadorj
and
pcuenca
authored
091fa42
Rm incorrect docstring
mishig25
committed
82311a5
Add FlaxModelMixin to __init__.py
mishig25
committed
a94968f
Merge branch 'main' into flax_model_mixin
Mishig Davaadorj
authored
e11c8e7
make fix-copies
mishig25
committed
642279f