Skip to content

GitLab

"docs/vscode:/vscode.git/clone" did not exist on "f7fae490980885e426fef01bb214025b9eddb832"
Switch branch/tag
Find file BlameHistoryPermalink
  • Pedro Cuenca's avatar
    Allow dtype to be specified in Flax pipeline (#600) · fb2fbab1
    Pedro Cuenca authored 
    * Fix typo in docstring.

* Allow dtype to be overridden on model load.

This may be a temporary solution until #567 is addressed.

* Create latents in float32

The denoising loop always computes the next step in float32, so this
would fail when using `bfloat16`.
    fb2fbab1
configuration_utils.py 20.1 KB