Merge 55c7571 into 107647a

open-mmlab · Sep 5, 2023 · d508640 · d508640
2 parents 107647a + 55c7571
commit d508640
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 9 deletions.
diff --git a/tests/test_apis/test_mmagic_inferencer.py b/tests/test_apis/test_mmagic_inferencer.py
@@ -52,10 +52,6 @@ def test_edit():
     assert result_img.shape == (4, 3, 32, 32)
 
 
-if __name__ == '__main__':
-    test_edit()
-
-
 def teardown_module():
     import gc
     gc.collect()

diff --git a/tests/test_models/test_diffusion_schedulers/test_ddim_scheduler.py b/tests/test_models/test_diffusion_schedulers/test_ddim_scheduler.py
@@ -48,10 +48,6 @@ def test_ddim_step():
     assert result['prev_sample'].shape == (1, 4, 32, 32)
 
 
-if __name__ == '__main__':
-    test_ddim_step()
-
-
 def teardown_module():
     import gc
     gc.collect()

diff --git a/tests/test_models/test_losses/test_clip_loss.py b/tests/test_models/test_losses/test_clip_loss.py
@@ -17,7 +17,7 @@
     digit_version(TORCH_VERSION) <= digit_version('1.6.0'),
     reason='version limitation')
 def test_clip_loss():
-    clip_loss = CLIPLoss(clip_model=dict(in_size=32))
+    clip_loss = CLIPLoss(clip_model=dict(in_size=32, clip_type='RN50'))
 
     image = torch.randn(1, 3, 32, 32)
     text = 'Image for test'