Skip to content

Commit

Permalink
Add hf_T5_generate to inline_inbuilt_nn_modules (#131804)
Browse files Browse the repository at this point in the history
Summary:
Fixes pytorch/pytorch#121989

We are turning on the flag by default in another PR. But that PR can go
through reverts. So, forcibly adding the benchmark to prevent dashboard
fluctuation in case of reverts.

X-link: pytorch/pytorch#131804
Approved by: https://github.com/yanboliang, https://github.com/shunting314
ghstack dependencies: #131795, #131801

Reviewed By: clee2000

Differential Revision: D60269692

Pulled By: anijain2305

fbshipit-source-id: d014844a90a621eb99ba511200b21c930a9f5638
  • Loading branch information
anijain2305 authored and facebook-github-bot committed Jul 26, 2024
1 parent 5cc1d01 commit 7797766
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions userbenchmark/dynamo/dynamobench/torchbench.py
Original file line number Diff line number Diff line change
Expand Up @@ -242,6 +242,7 @@ def inline_inbuilt_nn_modules_models(self):
"detectron2_maskrcnn_r_101_fpn",
"vision_maskrcnn",
"doctr_reco_predictor",
"hf_T5_generate",
}

def load_model(
Expand Down

0 comments on commit 7797766

Please sign in to comment.