[run-slow] Fixing minimum

Author: qubvel

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
 

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"