push

pytorch · Jul 13, 2023 · 80d36d2 · 80d36d2
1 parent 61b3c1f
commit 80d36d2
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/torchbenchmark/models/stable_diffusion/metadata.yaml b/torchbenchmark/models/stable_diffusion/metadata.yaml
@@ -0,0 +1,8 @@
+devices:
+  NVIDIA A100-SXM4-40GB:
+    eval_batch_size: 32
+eval_benchmark: false
+eval_deterministic: false
+eval_nograd: true
+train_benchmark: false
+train_deterministic: false