add flax requirement (#2894)

Co-authored-by: yiyixuxu <yixu310@gmail,com>

add flax requirement (#2894)
Co-authored-by: yiyixuxu <yixu310@gmail,com>
b3d5cc4a · YiYi Xu · GitHub · b2021273 · b3d5cc4a
Unverified Commit b3d5cc4a authored Mar 30, 2023 by YiYi Xu Committed by GitHub Mar 30, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 9 additions and 0 deletions

examples/controlnet/requirements_flax.txt examples/controlnet/requirements_flax.txt +9 -0

No files found.
--- a/examples/controlnet/requirements_flax.txt
+++ b/examples/controlnet/requirements_flax.txt
+transformers>=4.25.1
+datasets
+flax
+optax
+torch
+torchvision
+ftfy
+tensorboard
+Jinja2