Skip to content

Commit 88efcfe

Browse files
committed
fix
1 parent ad62caa commit 88efcfe

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

_unittests/ut_torch_export_patches/test_patch_serialization.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -181,7 +181,7 @@ def test_base_sliding_window_cache_unflatten_flatten(self):
181181
self.assertEqualAny([cache], cache2)
182182

183183
@ignore_warnings(UserWarning)
184-
@requires_torch("2.8")
184+
@requires_torch("2.7.99")
185185
def test_sliding_window_cache_export(self):
186186
class Model(torch.nn.Module):
187187
def forward(self, cache):
@@ -276,7 +276,7 @@ def forward(self, cache):
276276
torch.export.export(model, (bo,), dynamic_shapes=(ds,))
277277

278278
@ignore_warnings(UserWarning)
279-
@requires_torch("2.8")
279+
@requires_torch("2.7.99")
280280
def test_static_cache(self):
281281
bo = make_static_cache(
282282
[

0 commit comments

Comments
 (0)