Finally

pytorch · Dec 17, 2024 · 368caaf · 368caaf
1 parent 217720d
commit 368caaf
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/.github/workflows/_linux-benchmark-cuda.yml b/.github/workflows/_linux-benchmark-cuda.yml
@@ -47,7 +47,7 @@ jobs:
         with:
           python-version: "3.9"
 
-      - name: Install dependencies
+      - name: Install torch dependencies
         shell: bash
         working-directory: benchmark
         run: |

diff --git a/userbenchmark/torchao/run.py b/userbenchmark/torchao/run.py
@@ -29,7 +29,8 @@ def _get_ci_args(
         f"--{experiment}",
         "--output",
         f"{str(OUTPUT_DIR.joinpath(output_file_name).resolve())}",
-        "--only hf_T5",  # DEBUG: TO BE REMOVED
+        "--only",  # DEBUG: TO BE REMOVED
+        "hf_T5",  # DEBUG: TO BE REMOVED
     ]
     return ci_args