re-enable tests w change in patching

This commit is contained in:
Wing Lian
2024-12-08 14:52:09 -05:00
parent 1302e31049
commit 26bd81cec0
3 changed files with 3 additions and 4 deletions

View File

@@ -4,7 +4,7 @@ fix for FSDP optimizer save in trainer w 4.47.0
import inspect
import logging
from transformers.trainer import Trainer
from transformers import Trainer
from axolotl.monkeypatch.unsloth_ import detab_code

View File

@@ -5,8 +5,7 @@ see https://github.com/huggingface/transformers/pull/35128
import inspect
import logging
from transformers import LlamaForCausalLM
from transformers.trainer import Trainer
from transformers import LlamaForCausalLM, Trainer
from axolotl.monkeypatch.unsloth_ import detab_code