Skip to content

Commit 8cd92b5

Browse files
committed
small.
1 parent 7dfd599 commit 8cd92b5

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

tests/pipelines/test_pipelines.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1997,7 +1997,9 @@ def test_from_save_pretrained(self):
19971997
reason="Torch Dynamo isn't yet supported for Python 3.12.",
19981998
)
19991999
def test_from_save_pretrained_dynamo(self):
2000-
run_test_in_subprocess(test_case=self, target_func=_test_from_save_pretrained_dynamo, inputs=None)
2000+
torch.compiler.rest()
2001+
with torch._inductor.utils.fresh_inductor_cache():
2002+
run_test_in_subprocess(test_case=self, target_func=_test_from_save_pretrained_dynamo, inputs=None)
20012003

20022004
def test_from_pretrained_hub(self):
20032005
model_path = "google/ddpm-cifar10-32"

0 commit comments

Comments
 (0)