Skip to content

Commit acd6d19

Browse files
committed
fix: lint errors
Signed-off-by: Mehant Kammakomati <[email protected]>
1 parent 78123d6 commit acd6d19

File tree

2 files changed

+7
-2
lines changed

2 files changed

+7
-2
lines changed

plugins/accelerated-moe/src/fms_acceleration_moe/framework_plugin_scattermoe.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,11 @@ class ScatterMoEAccelerationPlugin(AccelerationPlugin):
4040
# if we decide to extract the kernels, then we do not need to anymore,
4141
# https://github.com/foundation-model-stack/fms-acceleration/issues/105
4242

43-
restricted_model_archs = ["GraniteMoeForCausalLM", "MixtralForCausalLM", "GraniteMoeSharedForCausalLM"]
43+
restricted_model_archs = [
44+
"GraniteMoeForCausalLM",
45+
"MixtralForCausalLM",
46+
"GraniteMoeSharedForCausalLM",
47+
]
4448

4549
def __init__(self, configurations: Dict[str, Dict]):
4650
super().__init__(configurations)

plugins/fused-ops-and-kernels/src/fms_acceleration_foak/models/granitemoeshared.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -108,7 +108,8 @@ def get_mp_rules(base_type: str):
108108
ModelPatcherRule(
109109
rule_id="granitemoeshared-rope",
110110
import_and_maybe_reload=(
111-
"transformers.models.granitemoeshared.modeling_granitemoeshared.apply_rotary_pos_emb",
111+
"transformers.models.granitemoeshared.\
112+
modeling_granitemoeshared.apply_rotary_pos_emb",
112113
fast_rope_embedding,
113114
None,
114115
),

0 commit comments

Comments
 (0)