diffusion multinode working

mila-iqia · Aug 9, 2024 · da4f8be · da4f8be
1 parent daf3612
commit da4f8be
Show file tree

Hide file tree

Showing 35 changed files with 1,504 additions and 1,116 deletions.
diff --git a/.pin/constraints-rocm-torch.txt b/.pin/constraints-rocm-torch.txt
diff --git a/.pin/constraints-xpu-torch.txt b/.pin/constraints-xpu-torch.txt
diff --git a/benchmarks/brax/requirements.in b/benchmarks/brax/requirements.in
@@ -1,5 +1,4 @@
-jax[cuda12]
---find-links https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
+jax
 torch
 brax
 voir>=0.2.10,<0.3