Skip to content

Commit

Permalink
check with only auto and vit
Browse files Browse the repository at this point in the history
  • Loading branch information
ydshieh committed Sep 3, 2024
1 parent e44de7d commit 9326cfb
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions tests/models/vit/test_modeling_vit.py
Original file line number Diff line number Diff line change
Expand Up @@ -212,6 +212,9 @@ def setUp(self):
self.model_tester = ViTModelTester(self)
self.config_tester = ConfigTester(self, config_class=ViTConfig, has_text_modality=False, hidden_size=37)

def test_foo(self):
assert 1 == 2

@unittest.skip(
"Since `torch==2.3+cu121`, although this test passes, many subsequent tests have `CUDA error: misaligned address`."
"If `nvidia-xxx-cu118` are also installed, no failure (even with `torch==2.3+cu121`)."
Expand Down

0 comments on commit 9326cfb

Please sign in to comment.