Merge branch 'master' into docs

mlcommons · Nov 19, 2024 · e66b13d · e66b13d
2 parents 6f4e17b + 4763c6b
commit e66b13d
Show file tree

Hide file tree

Showing 7 changed files with 13 additions and 345 deletions.
diff --git a/compliance/nvidia/README.md b/compliance/nvidia/README.md
@@ -30,12 +30,11 @@ The `run_verification.py` found in each test directory will copy the test files
 
 | model | Required Compliance Tests
 | ---- | ---- |
-| resnet50-v1.5 | [TEST01](./TEST01/), [TEST04](./TEST04/), [TEST05](./TEST05/) |
-| retinanet 800x800 | [TEST01](./TEST01/), [TEST05](./TEST05/) |
-| bert | [TEST01](./TEST01/), [TEST05](./TEST05/) |
-| dlrm-v2 | [TEST01](./TEST01/), [TEST05](./TEST05/) |
-| 3d-unet | [TEST01](./TEST01/), [TEST05](./TEST05/) |
-| rnnt | [TEST01](./TEST01/), [TEST05](./TEST05/) |
+| resnet50-v1.5 | [TEST01](./TEST01/), [TEST04](./TEST04/) |
+| retinanet 800x800 | [TEST01](./TEST01/) |
+| bert | [TEST01](./TEST01/) |
+| dlrm-v2 | [TEST01](./TEST01/) |
+| 3d-unet | [TEST01](./TEST01/) |
 | gpt-j | - |
 | stable-diffusion-xl | [TEST01](./TEST01/), [TEST04](./TEST04/) |
 | Llama2-70b | [TEST06](./TEST06/) |

diff --git a/compliance/nvidia/TEST05/README.md b/compliance/nvidia/TEST05/README.md
diff --git a/compliance/nvidia/TEST05/audit.config b/compliance/nvidia/TEST05/audit.config
diff --git a/compliance/nvidia/TEST05/run_verification.py b/compliance/nvidia/TEST05/run_verification.py
diff --git a/compliance/nvidia/TEST05/verify_performance.py b/compliance/nvidia/TEST05/verify_performance.py
diff --git a/text_to_image/requirements.txt b/text_to_image/requirements.txt
@@ -1,8 +1,8 @@
-diffusers==0.21.2
-transformers==4.33.2
-accelerate==0.23.0
-open-clip-torch==2.7.0
-opencv-python==4.8.1.78
+diffusers==0.30.3
+transformers==4.45.2
+accelerate==1.0.1
+open-clip-torch==2.26.1
+opencv-python==4.10.0.84
 pycocotools==2.0.7
-torchmetrics[image]==1.2.0
-scipy==1.9.1
+torchmetrics[image]==1.4.3
+scipy==1.10.1
diff --git a/tools/submission/submission_checker.py b/tools/submission/submission_checker.py
@@ -2906,7 +2906,7 @@ def check_compliance_dir(
         "llama2-70b-99.9",
         "stable-diffusion-xl",
         "mixtral-8x7b",
-    ]:
+    ] or config.version not in ["v4.0", "v4.1"]:
         test_list.remove("TEST05")
 
     if model in [