DeciLM-7B-instruct / version_check.py
tomer-deci's picture
added support for text-generation pipeline
49422ae
raw
history blame contribute delete
383 Bytes
import transformers
from packaging import version
MIN_VERSION = "4.35.2"
def check_transformers_version():
if version.parse(transformers.__version__) < version.parse(MIN_VERSION):
raise ImportError(
f"You are using transformers=={transformers.__version__}, but transformers>={MIN_VERSION} is required to use DeciLM. Please upgrade transformers."
)