Hugging Face's logo

Spaces:
flax-community
/
dalle-mini
Running

App Files Community
9
dalle-mini / dalle_mini /modeling_bart_flax.py

Commit History

fix(model): use correct params
a11892f

boris commited on

add property to get num params
e5a52b9

valhalla commited on

make checkpointing optional
f6c4cb2

valhalla commited on

handle dtype for embeddings
29db327

valhalla commited on

add gradient checkpointing
95a8ed2

valhalla commited on

fix layernorm
7774483

valhalla commited on

remove bias and minor fixes
180ed1e

valhalla commited on

add standalone modeling file
6197b2f

valhalla commited on