diffusers
[Flax] added broadcast_to_shape_from_left helper and Scheduler tests
#864
Merged
Merged
[Flax] added broadcast_to_shape_from_left helper and Scheduler tests #864
patrickvonplaten
merged
15 commits into
huggingface:main
from
kashif:jax-broadcast
added broadcast_to_shape_from_left helper
3f4326df
patrickvonplaten
approved these changes
on 2022-10-17
initial tests
f80717ae
fixed pndm tests
6781755c
kashif
changed the title
Merge branch 'main' into jax-broadcast
c4abce51
shape required for pndm
b00f754b
Merge branch 'jax-broadcast' of https://github.com/kashif/diffusers i…
f9f25bef
added require_flax
ce2e7b88
fix style
3f2f4f97
fix more imports
af1f84d1
patrickvonplaten
commented
on 2022-10-19
Merge branch 'main' into jax-broadcast
518c2b93
Merge branch 'main' into jax-broadcast
d6616ad0
Merge branch 'main' into jax-broadcast
275d3413
Merge branch 'main' into jax-broadcast
1a5e0259
Merge branch 'main' into jax-broadcast
e46cf63c
Merge branch 'main' into jax-broadcast
5289b6a0
Login to write a write a comment.
Login via GitHub