From 43d48dec8cd7b055ff9f93da5db6234f7b552f34 Mon Sep 17 00:00:00 2001 From: Jane Xu Date: Thu, 28 Sep 2023 09:19:03 -0700 Subject: [PATCH] add new model hf_clip to DENSE_MODELS (#1943) Summary: Fixes https://github.com/pytorch/benchmark/issues/1942 Pull Request resolved: https://github.com/pytorch/benchmark/pull/1943 Reviewed By: xuzhao9 Differential Revision: D49733689 Pulled By: janeyx99 fbshipit-source-id: 241b1674b07cfdda69f382a0e8420c8440d4df12 --- userbenchmark/optim/run.py | 1 + 1 file changed, 1 insertion(+) diff --git a/userbenchmark/optim/run.py b/userbenchmark/optim/run.py index 0d11b956aa..f0f0a384dd 100644 --- a/userbenchmark/optim/run.py +++ b/userbenchmark/optim/run.py @@ -187,6 +187,7 @@ def get_unstable_models() -> Set[str]: 'hf_T5_generate', 'hf_T5_large', 'hf_Whisper', + 'hf_clip', 'lennard_jones', 'llama', 'llama_v2_7b_16h',