check with only auto and vit

huggingface · Sep 3, 2024 · e44de7d · e44de7d
1 parent ee5fc1d
commit e44de7d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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/auto"], ["models/vit"]]
     print(model_splits)