lint: remove unused import

Signed-off-by: Will Johnson <[email protected]>
foundation-model-stack · Jan 7, 2025 · c1e866a · c1e866a
1 parent b154026
commit c1e866a
Showing 1 changed file with 1 addition and 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
@@ -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