Merge branch 'open-mmlab:main' into main

liuwenran · Aug 21, 2023 · 0fb4a67 · 0fb4a67
2 parents 3d8740a + 309ad75
commit 0fb4a67
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/.circleci/test.yml b/.circleci/test.yml
@@ -63,6 +63,7 @@ jobs:
             pip install -U openmim
             mim install 'mmcv >= 2.0.0'
             pip install -r requirements/tests.txt
+            pip install scipy==1.7.2
       - run:
           name: Build and install
           command: |
@@ -106,6 +107,7 @@ jobs:
             docker exec mmagic pip install -U openmim
             docker exec mmagic mim install 'mmcv >= 2.0.0'
             docker exec mmagic pip install -r requirements/tests.txt
+            docker exec mmagic pip install 'scipy == 1.7.2'
       - run:
           name: Build and install
           command: |

diff --git a/tests/test_models/test_archs/test_wrapper.py b/tests/test_models/test_archs/test_wrapper.py
@@ -40,7 +40,10 @@ def test_build(self):
         self.assertIn(f'From Config: {config_path}', model_str)
 
         # 2. test save as diffuser
-        model.save_pretrained(model_path)
+        if digit_version(TORCH_VERSION) < digit_version('2.0.1'):
+            model.save_pretrained(model_path, safe_serialization=False)
+        else:
+            model.save_pretrained(model_path)
 
         # 3. test from_pretrained
         model = MODELS.build(