Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
renzhc
diffusers_dcu
Repository
6043838971194b855d94aaf8fdf18efcd0e41776
Switch branch/tag
diffusers_dcu
src
diffusers
schedulers
scheduling_ddpm_flax.py
Find file
Blame
History
Permalink
Fix OOM when using PyTorch with JAX installed. (#1795)
· 60438389
Pedro Cuenca
authored
Dec 21, 2022
Don't initialize Jax on startup.
60438389
scheduling_ddpm_flax.py
12.4 KB
Edit
Web IDE
Replace scheduling_ddpm_flax.py
×
Attach a file by drag & drop or
click to upload
Commit message
Replace scheduling_ddpm_flax.py
Replace file
Cancel
A new branch will be created in your fork and a new merge request will be started.
