Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add
cleanup_torchscript_cache
to test_quantize_conformance.py (#2171)
### Changes Add `cleanup_torchscript_cache` function to test_quantize_conformance.py ### Reason for changes After run torch.jit.trace in convert_model, PyTorch does not clear the trace cache automatically. Same function in torch test and openvino notebok: https://github.com/pytorch/pytorch/blob/main/torch/testing/_internal/jit_utils.py#L59 https://github.com/openvinotoolkit/openvino_notebooks/blob/1932d4b4e99116bdedaa620c9dc92069fbb1f05e/notebooks/236-stable-diffusion-v2/implementation/conversion_helper_utils.py#L8
- Loading branch information