Former-commit-id: 5665062ca0bfb166cd8f2e896e2b0970037373f6
This commit is contained in:
hiyouga 2024-07-22 21:10:15 +08:00
parent 37c6a0c6dc
commit 1cddf80a97

View File

@ -21,7 +21,6 @@ from peft import PeftModel
from transformers import PreTrainedModel, PreTrainedTokenizerBase, is_torch_npu_available
from transformers.integrations import is_deepspeed_zero3_enabled
from transformers.modeling_utils import is_fsdp_enabled
from transformers.utils.versions import require_version
from ..extras.logging import get_logger
from ..extras.misc import infer_optim_dtype