Fix OwlViT tests (#18253)

* Fix OwlViT tests * Forgot one
huggingface · Jul 22, 2022 · 06d98e2 · 06d98e2
1 parent 12d66b4
commit 06d98e2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/transformers/models/owlvit/__init__.py b/src/transformers/models/owlvit/__init__.py
@@ -40,7 +40,7 @@
 
 
 try:
-    if not is_vision_available() or not is_torch_available():
+    if not is_vision_available():
         raise OptionalDependencyNotAvailable()
 except OptionalDependencyNotAvailable:
     pass
@@ -72,7 +72,7 @@
     from .processing_owlvit import OwlViTProcessor
 
     try:
-        if not is_vision_available() or not is_torch_available():
+        if not is_vision_available():
             raise OptionalDependencyNotAvailable()
     except OptionalDependencyNotAvailable:
         pass