diff --git a/torchbenchmark/models/stable_diffusion/metadata.yaml b/torchbenchmark/models/stable_diffusion/metadata.yaml new file mode 100644 index 0000000000..d4f9a26cfe --- /dev/null +++ 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