From fbbf2a5c57bd7f4230ec8763bc1c0b6752d25bf1 Mon Sep 17 00:00:00 2001
From: ydshieh <ydshieh@users.noreply.github.com>
Date: Thu, 6 Jun 2024 15:49:43 +0200
Subject: [PATCH] run fix

---
 .github/workflows/self-scheduled-caller.yml | 34 +--------------------
 utils/split_model_tests.py                  |  1 +
 2 files changed, 2 insertions(+), 33 deletions(-)

diff --git a/.github/workflows/self-scheduled-caller.yml b/.github/workflows/self-scheduled-caller.yml
index 40689c629a09bf..2431758ae7c6fc 100644
--- a/.github/workflows/self-scheduled-caller.yml
+++ b/.github/workflows/self-scheduled-caller.yml
@@ -7,7 +7,7 @@ on:
     - cron: "17 2 * * *"
   push:
     branches:
-      - run_scheduled_ci*
+      - fix_main_ci
 
 jobs:
   model-ci:
@@ -18,38 +18,6 @@ jobs:
       slack_report_channel: "#transformers-ci-daily-models"
     secrets: inherit
 
-  torch-pipeline:
-    name: Torch pipeline CI
-    uses: ./.github/workflows/self-scheduled.yml
-    with:
-      job: run_pipelines_torch_gpu
-      slack_report_channel: "#transformers-ci-daily-pipeline-torch"
-    secrets: inherit
-
-  tf-pipeline:
-    name: TF pipeline CI
-    uses: ./.github/workflows/self-scheduled.yml
-    with:
-      job: run_pipelines_tf_gpu
-      slack_report_channel: "#transformers-ci-daily-pipeline-tf"
-    secrets: inherit
-
-  example-ci:
-    name: Example CI
-    uses: ./.github/workflows/self-scheduled.yml
-    with:
-      job: run_examples_gpu
-      slack_report_channel: "#transformers-ci-daily-examples"
-    secrets: inherit
-
-  deepspeed-ci:
-    name: DeepSpeed CI
-    uses: ./.github/workflows/self-scheduled.yml
-    with:
-      job: run_torch_cuda_extensions_gpu
-      slack_report_channel: "#transformers-ci-daily-deepspeed"
-    secrets: inherit
-
   quantization-ci:
     name: Quantization CI
     uses: ./.github/workflows/self-scheduled.yml
diff --git a/utils/split_model_tests.py b/utils/split_model_tests.py
index e5083aaeb46fa5..d7b1337eb01ea5 100644
--- a/utils/split_model_tests.py
+++ b/utils/split_model_tests.py
@@ -62,4 +62,5 @@
         start = end
         end = start + num_jobs_per_splits + (1 if idx < num_jobs % args.num_splits else 0)
         model_splits.append(d[start:end])
+    model_splits = [["models/vit"], ["models/levit"]]
     print(model_splits)