Fix OOM when using PyTorch with JAX installed. (#1795) · addianto/diffusers@6043838 · GitHub
Skip to content

Commit 6043838

Browse files
pcuencapcuenca
authored
Fix OOM when using PyTorch with JAX installed. (huggingface#1795)
Don't initialize Jax on startup.
1 parent 4125756 commit 6043838

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

‎src/diffusers/schedulers/scheduling_ddpm_flax.py‎

Lines changed: 1 addition & 1 deletion

0 commit comments

Comments
 (0)