[CI] Add DALLE2_pytorch to FORCE_AMP_FOR_FP16_BF16_MODELS

ghstack-source-id: 312f2649c10ad07c8e56dfe29004f9bc5ded9b2d Pull Request resolved: #104283
pytorch · Jun 27, 2023 · f3217e1 · f3217e1
1 parent ab95770
commit f3217e1
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/benchmarks/dynamo/torchbench.py b/benchmarks/dynamo/torchbench.py
@@ -211,6 +211,7 @@ def setup_torchbench_cwd():
 }
 
 FORCE_AMP_FOR_FP16_BF16_MODELS = {
+    "DALLE2_pytorch",
     "doctr_det_predictor",
     "doctr_reco_predictor",
     "Super_SloMo",