From 34ce94db424445b38eb56a6c842e55a2122d4a9d Mon Sep 17 00:00:00 2001
From: YunLiu <55491388+KumoLiu@users.noreply.github.com>
Date: Tue, 13 Aug 2024 14:44:53 +0800
Subject: [PATCH] Fix ci issue in test_vit (#8013)

Fixes #8012


### Types of changes
<!--- Put an `x` in all the boxes that apply, and remove the not
applicable items -->
- [x] Non-breaking change (fix or new feature that would not break
existing functionality).
- [ ] Breaking change (fix or new feature that would cause existing
functionality to change).
- [ ] New tests added to cover the changes.
- [ ] Integration tests passed locally by running `./runtests.sh -f -u
--net --coverage`.
- [ ] Quick tests passed locally by running `./runtests.sh --quick
--unittests --disttests`.
- [ ] In-line docstrings updated.
- [ ] Documentation updated, tested `make html` command in the `docs/`
folder.

---------

Signed-off-by: YunLiu <55491388+KumoLiu@users.noreply.github.com>
---
 requirements-dev.txt | 1 -
 tests/test_vit.py    | 2 +-
 2 files changed, 1 insertion(+), 2 deletions(-)

diff --git a/requirements-dev.txt b/requirements-dev.txt
index 76f1952345..9aad0804e6 100644
--- a/requirements-dev.txt
+++ b/requirements-dev.txt
@@ -58,5 +58,4 @@ lpips==0.1.4
 nvidia-ml-py
 huggingface_hub
 pyamg>=5.0.0
-git+https://github.com/Project-MONAI/GenerativeModels.git@7428fce193771e9564f29b91d29e523dd1b6b4cd
 git+https://github.com/facebookresearch/segment-anything.git@6fdee8f2727f4506cfbbe553e23b895e27956588
diff --git a/tests/test_vit.py b/tests/test_vit.py
index d638c0116a..a3ffd0b2ef 100644
--- a/tests/test_vit.py
+++ b/tests/test_vit.py
@@ -106,7 +106,7 @@ def test_ill_arg(
             )
 
     @parameterized.expand(TEST_CASE_Vit[:1])
-    @SkipIfBeforePyTorchVersion((1, 9))
+    @SkipIfBeforePyTorchVersion((2, 0))
     def test_script(self, input_param, input_shape, _):
         net = ViT(**(input_param))
         net.eval()