[run-slow] Fixing minimum

qubvel-org · Dec 23, 2024 · 63ba1c4 · 63ba1c4
1 parent 238dd11
commit 63ba1c4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/tests/models/base.py b/tests/models/base.py
@@ -137,7 +137,7 @@ def test_classification_head(self):
 
         self.assertEqual(cls_probs.shape[1], 10)
 
-    @requires_torch_greater_or_equal("2.0.0")
+    @requires_torch_greater_or_equal("2.0.1")
     def test_save_load_with_hub_mixin(self):
         # instantiate model
         model = smp.create_model(
@@ -173,7 +173,7 @@ def test_save_load_with_hub_mixin(self):
         self.assertIn("my_awesome_metric", readme)
 
     @slow_test
-    @requires_torch_greater_or_equal("2.0.0")
+    @requires_torch_greater_or_equal("2.0.1")
     def test_preserve_forward_output(self):
         from huggingface_hub import hf_hub_download
 

diff --git a/tests/models/test_segformer.py b/tests/models/test_segformer.py
@@ -11,7 +11,7 @@ class TestSegformerModel(base.BaseModelTester):
     test_model_type = "segformer"
 
     @slow_test
-    @requires_torch_greater_or_equal("2.0.0")
+    @requires_torch_greater_or_equal("2.0.1")
     def test_load_pretrained(self):
         hub_checkpoint = "smp-hub/segformer-b0-512x512-ade-160k"