[build-ci-image]

huggingface · May 22, 2024 · 549ccc6 · 549ccc6
1 parent 782df71
commit 549ccc6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -173,7 +173,7 @@
     "tensorflow-cpu>2.9,<2.16",
     "tensorflow>2.9,<2.16",
     "tensorflow-text<2.16",
-    "tensorflow-probability<2.16",
+    "tensorflow-probability<0.24",
     "tf2onnx",
     "timeout-decorator",
     "timm<=0.9.16",

diff --git a/utils/tests_fetcher.py b/utils/tests_fetcher.py
@@ -93,6 +93,7 @@
     "vit",
     "whisper",
     # Pipeline-specific model (to be sure each pipeline has one model in this list)
+    "tapas",
     "vilt",
     "clap",
     "detr",
-Original file line number
+Diff line change
@@ Expand Up / @@ -93,6 +93,7 @@ @@
         "vit",
         "whisper",
         # Pipeline-specific model (to be sure each pipeline has one model in this list)
+        "tapas",
         "vilt",
         "clap",
         "detr",
@@ Expand Down @@