from packaging import version
import transformers

_TRANSFORMERS_VERSION = version.parse(transformers.__version__)

if _TRANSFORMERS_VERSION >= version.parse("5.0.0"):
    from transformers.models.llama.configuration_llama import LlamaConfig
else:
    from .configuration_llama_patch_4x import LlamaConfig