Merge branch 'master' into more_dockers

mila-iqia · Aug 23, 2024 · 723574e · 723574e
2 parents 230ae61 + bdf7705
commit 723574e
Show file tree

Hide file tree

Showing 75 changed files with 101 additions and 88 deletions.
diff --git a/.pin/constraints-cuda-torch.txt b/.pin/constraints-cuda-torch.txt
diff --git a/.pin/constraints-hpu-torch.txt b/.pin/constraints-hpu-torch.txt
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/_templates/simple/requirements.in b/benchmarks/_templates/simple/requirements.in
@@ -1,2 +1,2 @@
-voir>=0.2.17,<0.3
+voir>=0.2.19,<0.3
 torch
diff --git a/benchmarks/_templates/stdout/requirements.in b/benchmarks/_templates/stdout/requirements.in
@@ -1,2 +1,2 @@
-voir>=0.2.17,<0.3
+voir>=0.2.19,<0.3
 torch
diff --git a/benchmarks/_templates/voir/requirements.in b/benchmarks/_templates/voir/requirements.in
@@ -1,2 +1,2 @@
-voir>=0.2.17,<0.3
+voir>=0.2.19,<0.3
 torch
diff --git a/benchmarks/accelerate_opt/requirements.cuda.txt b/benchmarks/accelerate_opt/requirements.cuda.txt
diff --git a/benchmarks/accelerate_opt/requirements.hpu.txt b/benchmarks/accelerate_opt/requirements.hpu.txt
diff --git a/benchmarks/accelerate_opt/requirements.rocm.txt b/benchmarks/accelerate_opt/requirements.rocm.txt
diff --git a/benchmarks/accelerate_opt/requirements.xpu.txt b/benchmarks/accelerate_opt/requirements.xpu.txt
diff --git a/benchmarks/brax/requirements.cuda.txt b/benchmarks/brax/requirements.cuda.txt
diff --git a/benchmarks/brax/requirements.hpu.txt b/benchmarks/brax/requirements.hpu.txt
diff --git a/benchmarks/brax/requirements.in b/benchmarks/brax/requirements.in
@@ -2,4 +2,4 @@ jax[cuda12]
 --find-links https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
 torch
 brax
-voir>=0.2.10,<0.3
+voir>=0.2.19,<0.3
diff --git a/benchmarks/brax/requirements.rocm.txt b/benchmarks/brax/requirements.rocm.txt
diff --git a/benchmarks/brax/requirements.xpu.txt b/benchmarks/brax/requirements.xpu.txt
diff --git a/benchmarks/diffusion/requirements.cuda.txt b/benchmarks/diffusion/requirements.cuda.txt
diff --git a/benchmarks/diffusion/requirements.in b/benchmarks/diffusion/requirements.in
@@ -1,4 +1,4 @@
-voir>=0.2.17,<0.3
+voir>=0.2.19,<0.3
 diffusers
 diffusers[torch]
 accelerate

diff --git a/benchmarks/dinov2/requirements.cuda.txt b/benchmarks/dinov2/requirements.cuda.txt
diff --git a/benchmarks/dinov2/requirements.in b/benchmarks/dinov2/requirements.in
@@ -1,4 +1,4 @@
-voir>=0.2.17,<0.3
+voir>=0.2.19,<0.3
 torch
 xformers
 torchvision

diff --git a/benchmarks/dlrm/requirements.cuda.txt b/benchmarks/dlrm/requirements.cuda.txt
diff --git a/benchmarks/dlrm/requirements.hpu.txt b/benchmarks/dlrm/requirements.hpu.txt
diff --git a/benchmarks/dlrm/requirements.rocm.txt b/benchmarks/dlrm/requirements.rocm.txt
diff --git a/benchmarks/dlrm/requirements.xpu.txt b/benchmarks/dlrm/requirements.xpu.txt
diff --git a/benchmarks/flops/requirements.cuda.txt b/benchmarks/flops/requirements.cuda.txt
diff --git a/benchmarks/flops/requirements.hpu.txt b/benchmarks/flops/requirements.hpu.txt
diff --git a/benchmarks/flops/requirements.rocm.txt b/benchmarks/flops/requirements.rocm.txt
diff --git a/benchmarks/flops/requirements.xpu.txt b/benchmarks/flops/requirements.xpu.txt
diff --git a/benchmarks/huggingface/requirements.cuda.txt b/benchmarks/huggingface/requirements.cuda.txt
diff --git a/benchmarks/huggingface/requirements.hpu.txt b/benchmarks/huggingface/requirements.hpu.txt
diff --git a/benchmarks/huggingface/requirements.rocm.txt b/benchmarks/huggingface/requirements.rocm.txt
diff --git a/benchmarks/huggingface/requirements.xpu.txt b/benchmarks/huggingface/requirements.xpu.txt
diff --git a/benchmarks/lightning/requirements.cuda.txt b/benchmarks/lightning/requirements.cuda.txt
diff --git a/benchmarks/llama/requirements.cuda.txt b/benchmarks/llama/requirements.cuda.txt
diff --git a/benchmarks/llama/requirements.hpu.txt b/benchmarks/llama/requirements.hpu.txt
diff --git a/benchmarks/llama/requirements.rocm.txt b/benchmarks/llama/requirements.rocm.txt
diff --git a/benchmarks/llama/requirements.xpu.txt b/benchmarks/llama/requirements.xpu.txt
diff --git a/benchmarks/llm/prepare.py b/benchmarks/llm/prepare.py
@@ -57,7 +57,7 @@ def main():
     ignore_pattern = "*.safetensors"
 
     if config.get("safetensors", False):
-        ignore_pattern = "consolidated.*.pth"
+        ignore_pattern = "*consolidated.*.pth"
 
     download_args = [
         "download",

diff --git a/benchmarks/llm/requirements.cuda.txt b/benchmarks/llm/requirements.cuda.txt
diff --git a/benchmarks/llm/requirements.in b/benchmarks/llm/requirements.in
@@ -1,4 +1,4 @@
-voir>=0.2.17,<0.3
+voir>=0.2.19,<0.3
 torchtune
 torch
 PyYAML

diff --git a/benchmarks/rwkv/requirements.cuda.txt b/benchmarks/rwkv/requirements.cuda.txt
diff --git a/benchmarks/rwkv/requirements.hpu.txt b/benchmarks/rwkv/requirements.hpu.txt
diff --git a/benchmarks/rwkv/requirements.in b/benchmarks/rwkv/requirements.in
@@ -3,5 +3,5 @@ torch
 deepspeed
 pytorch-lightning<2.0
 ninja
-voir>=0.2.10,<0.3
+voir>=0.2.19,<0.3
 pydantic<2
diff --git a/benchmarks/rwkv/requirements.rocm.txt b/benchmarks/rwkv/requirements.rocm.txt
diff --git a/benchmarks/rwkv/requirements.xpu.txt b/benchmarks/rwkv/requirements.xpu.txt
diff --git a/benchmarks/stargan/requirements.cuda.txt b/benchmarks/stargan/requirements.cuda.txt
diff --git a/benchmarks/stargan/requirements.hpu.txt b/benchmarks/stargan/requirements.hpu.txt