Skip to content

Commit 7b63cd3

Browse files
authored
Apply suggestion from @rkazants
1 parent c036d9c commit 7b63cd3

1 file changed

Lines changed: 0 additions & 9 deletions

File tree

optimum/exporters/openvino/model_patcher.py

Lines changed: 0 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1744,15 +1744,6 @@ def __enter__(self):
17441744
"long_mscale", None
17451745
)
17461746

1747-
if is_transformers_version("<", "5"):
1748-
for layer in self._model.model.layers:
1749-
layer.block_sparse_moe._orig_forward = layer.block_sparse_moe.forward
1750-
layer.block_sparse_moe.forward = types.MethodType(
1751-
_phi_moe_sparse_moe_block_forward, layer.block_sparse_moe
1752-
)
1753-
else:
1754-
self._model.set_experts_implementation("batched_mm")
1755-
17561747
def __exit__(self, exc_type, exc_value, traceback):
17571748
super().__exit__(exc_type, exc_value, traceback)
17581749

0 commit comments

Comments
 (0)