remove debugging code

huggingface · Aug 8, 2024 · 2d20a45 · 2d20a45
1 parent 104feb2
commit 2d20a45
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 10 deletions.
diff --git a/.github/workflows/update_llm_perf_cuda_pytorch.yaml b/.github/workflows/update_llm_perf_cuda_pytorch.yaml
@@ -1,12 +1,9 @@
 name: Update LLM Perf Benchmarks - CUDA PyTorch
 
-on: workflow_dispatch
-  # push:
-  #   branches:
-  #     - "add-t4-for-llm-perf-leaderboard"
-  # For debugging purpose the workflow is triggered manually
-  # schedule:
-  #   - cron: "0 */6 * * *"
+on: 
+  workflow_dispatch:  # Manual trigger
+  release:            # Trigger on new release
+    types: [published]  
 
 concurrency:
   cancel-in-progress: true
@@ -23,7 +20,7 @@ jobs:
         subset: [unquantized, bnb, awq, gptq]
 
         machine: [
-        # {name: 1xA10, runs-on: {group: 'aws-g5-4xlarge-plus'}}, 
+          {name: 1xA10, runs-on: {group: 'aws-g5-4xlarge-plus'}}, 
           {name: 1xT4, runs-on: {group: 'aws-g4dn-2xlarge'}}
         ]
 

diff --git a/llm_perf/utils.py b/llm_perf/utils.py
@@ -44,8 +44,8 @@
 #     model for model in PRETRAINED_OPEN_LLM_LIST if model.split("/")[0] in CANONICAL_ORGANIZATIONS
 # ]
 CANONICAL_PRETRAINED_OPEN_LLM_LIST = [
-    # "01-ai/Yi-6B",
-    # "01-ai/Yi-34B",
+    "01-ai/Yi-6B",
+    "01-ai/Yi-34B",
     "Deci/DeciLM-7B",
     "Deci/DeciCoder-1b",
     "EleutherAI/gpt-j-6b",