Fix hf_T5_generate

pytorch · Jan 28, 2025 · 28fa2a2 · 28fa2a2
1 parent 9396144
commit 28fa2a2
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/torchbenchmark/models/hf_T5_generate/metadata.yaml b/torchbenchmark/models/hf_T5_generate/metadata.yaml
@@ -6,3 +6,7 @@ eval_deterministic: false
 eval_nograd: true
 train_benchmark: false
 train_deterministic: false
+not_implemented:
+  # Timeout and exceed 300s time limit on A10G
+  - device: NVIDIA A10G
+    test: eval