additional linting

foundation-model-stack · Jul 28, 2024 · 1d498e0 · 1d498e0
1 parent cd253b3
commit 1d498e0
Show file tree

Hide file tree

Showing 4 changed files with 1 addition and 3 deletions.
diff --git a/plugins/fused-ops-and-kernels/src/fms_acceleration_foak/models/llama.py b/plugins/fused-ops-and-kernels/src/fms_acceleration_foak/models/llama.py
@@ -22,7 +22,6 @@
     LlamaRMSNorm,
 )
 from fms_acceleration.model_patcher import (
-    ModelPatcher,
     ModelPatcherRule,
     ModelPatcherTrigger,
     combine_functions,

diff --git a/plugins/fused-ops-and-kernels/src/fms_acceleration_foak/models/mistral.py b/plugins/fused-ops-and-kernels/src/fms_acceleration_foak/models/mistral.py
@@ -22,7 +22,6 @@
     MistralRMSNorm,
 )
 from fms_acceleration.model_patcher import (
-    ModelPatcher,
     ModelPatcherRule,
     ModelPatcherTrigger,
     combine_functions,

diff --git a/plugins/fused-ops-and-kernels/src/fms_acceleration_foak/models/mixtral.py b/plugins/fused-ops-and-kernels/src/fms_acceleration_foak/models/mixtral.py
@@ -21,7 +21,6 @@
     MixtralRMSNorm,
 )
 from fms_acceleration.model_patcher import (
-    ModelPatcher,
     ModelPatcherRule,
     ModelPatcherTrigger,
     combine_functions,

diff --git a/tox.ini b/tox.ini
@@ -21,6 +21,7 @@ allowlist_externals = bash
 description = run benchmarks 
 skip_install = true
 deps = 
+    matplotlib # this is for plotting benchmark comparisons in compare_with_reference.py
     packaging # this is required for flash-attn dep as fms_hf_tuning did not specify
     -e {toxinidir}/plugins/framework # install the framework here as the flash attention deps requires torch
 passenv = * # will pass the parent env, otherwise there are too many envs e.g. TRANSFORMERS that need to be set