[CI] Add DALLE2_pytorch to FORCE_AMP_FOR_FP16_BF16_MODELS

[ghstack-poisoned]
pytorch · Jun 27, 2023 · 344d259 · 344d259
1 parent ab95770
commit 344d259
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",