tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

chore: remove unused log

Browse Source
This commit is contained in:
NanoCode012
2025-03-31 16:20:15 +07:00
parent 873385b7d5
commit 7888a35118
1 changed files with 0 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/axolotl/monkeypatch/llama_attn_hijack_xformers.py
View File

@@ -4,7 +4,6 @@ Hijack the LlamaAttention forward method to use xformers if available.
Updated for transformers v4.50.0. Updated for transformers v4.50.0.
""" """
import logging
from typing import Optional from typing import Optional
import torch import torch
@@ -112,5 +111,3 @@ def hijack_llama_attention():
# Apply the patch # Apply the patch
llama_modeling.LlamaAttention.forward = patched_forward llama_modeling.LlamaAttention.forward = patched_forward
logging.info("Successfully patched LlamaAttention with xformers")