TildeOpen-30b-64k / configuration_llama.py
Martins Kronis
add YaRN patch for transformers 4x
a66e9c6
Raw
History Blame Contribute Delete
306 Bytes
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