We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7dfd599 commit 8cd92b5Copy full SHA for 8cd92b5
tests/pipelines/test_pipelines.py
@@ -1997,7 +1997,9 @@ def test_from_save_pretrained(self):
1997
reason="Torch Dynamo isn't yet supported for Python 3.12.",
1998
)
1999
def test_from_save_pretrained_dynamo(self):
2000
- run_test_in_subprocess(test_case=self, target_func=_test_from_save_pretrained_dynamo, inputs=None)
+ 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)
2003
2004
def test_from_pretrained_hub(self):
2005
model_path = "google/ddpm-cifar10-32"
0 commit comments