From c1e866a56a4ac2480a6139795d092d72d9f61ee7 Mon Sep 17 00:00:00 2001 From: Will Johnson Date: Tue, 7 Jan 2025 15:04:59 -0500 Subject: [PATCH] lint: remove unused import Signed-off-by: Will Johnson --- .../src/fms_acceleration_moe/framework_plugin_scattermoe.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/plugins/accelerated-moe/src/fms_acceleration_moe/framework_plugin_scattermoe.py b/plugins/accelerated-moe/src/fms_acceleration_moe/framework_plugin_scattermoe.py index 9fae7d5..b6a3c78 100644 --- a/plugins/accelerated-moe/src/fms_acceleration_moe/framework_plugin_scattermoe.py +++ b/plugins/accelerated-moe/src/fms_acceleration_moe/framework_plugin_scattermoe.py @@ -17,7 +17,7 @@ # Third Party from fms_acceleration import AccelerationPlugin -from transformers import AutoModelForCausalLM, TrainingArguments +from transformers import TrainingArguments from peft import LoraConfig import torch