Skip to content

Commit

Permalink
remove phimoe now that it is in transformers
Browse files Browse the repository at this point in the history
  • Loading branch information
winglian committed Nov 14, 2024
1 parent 783945c commit 9e2e8f7
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions src/axolotl/monkeypatch/multipack.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,8 +33,6 @@ def patch_for_multipack(model_type, model_name=None, is_remote_code=False):
patch_remote(model_name, ".configuration_gemmoe", ".modeling_gemmoe")
elif model_type == "deepseek_v2":
patch_remote(model_name, ".configuration_deepseek", ".modeling_deepseek")
elif model_type == "phimoe":
patch_remote(model_name, ".configuration_phimoe", ".modeling_phimoe")
elif hasattr(transformers, "modeling_flash_attention_utils") and not is_remote_code:
transformers.modeling_flash_attention_utils._get_unpad_data = ( # pylint: disable=protected-access
get_unpad_data
Expand Down

0 comments on commit 9e2e8f7

Please sign in to comment.