From a12941306a21ea2def6874de32ca11dfa59e2451 Mon Sep 17 00:00:00 2001 From: Kevin Wu Date: Tue, 25 Jun 2024 21:16:22 -0500 Subject: [PATCH] Fix formatting for run_transformers --- tools/run_transformers.py | 1 - 1 file changed, 1 deletion(-) diff --git a/tools/run_transformers.py b/tools/run_transformers.py index ca8b9875..aceed777 100644 --- a/tools/run_transformers.py +++ b/tools/run_transformers.py @@ -34,7 +34,6 @@ def run_model( to_profile: bool, device=None, ): - text_modules = [ AutoModelForQuestionAnswering, AutoModelForCausalLM,