Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 3 additions & 2 deletions test/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -620,11 +620,12 @@ def revert_liger_kernel_to_llava(model_config: MiniModelConfig):
Revert all Liger kernel patches applied to llava.
"""

from transformers.models.clip import modeling_clip
from transformers.models.llama import modeling_llama
from transformers.models.llava import modeling_llava

importlib.reload(modeling_clip)
# Note: Do NOT reload modeling_clip as it breaks CLIPVisionModel's
# output_hidden_states functionality in transformers v5.
# Liger kernel does not patch modeling_clip when model=None.
importlib.reload(modeling_llava)
importlib.reload(modeling_llama)

Expand Down