diff --git a/torchbenchmark/models/sam_fast/metadata.yaml b/torchbenchmark/models/sam_fast/metadata.yaml index 91e2f46c9e..632c63608f 100644 --- a/torchbenchmark/models/sam_fast/metadata.yaml +++ b/torchbenchmark/models/sam_fast/metadata.yaml @@ -14,3 +14,4 @@ not_implemented: # https://github.com/pytorch/benchmark/actions/runs/8972426901/job/24640395990 - device: NVIDIA A10G test: eval +skip_cuda_memory_leak: true