Unverified Commit c2283310 authored by Suraj Patil's avatar Suraj Patil Committed by GitHub
Browse files

[examples] add check_min_version (#1550)



* add check_min_version for examples

* move __version__ to the top

* Apply suggestions from code review
Co-authored-by: default avatarPedro Cuenca <pedro@huggingface.co>

* fix comment

* fix error_message

* adapt the install message
Co-authored-by: default avatarPedro Cuenca <pedro@huggingface.co>
parent ae4112d2
......@@ -15,6 +15,9 @@
import os
from packaging import version
from .. import __version__
from .deprecation_utils import deprecate
from .import_utils import (
ENV_VARS_TRUE_AND_AUTO_VALUES,
......@@ -88,3 +91,16 @@ _COMPATIBLE_STABLE_DIFFUSION_SCHEDULERS = [
"EulerAncestralDiscreteScheduler",
"DPMSolverMultistepScheduler",
]
def check_min_version(min_version):
if version.parse(__version__) < version.parse(min_version):
if "dev" in min_version:
error_message = (
"This example requires a source install from HuggingFace diffusers (see "
"`https://huggingface.co/docs/diffusers/installation#install-from-source`),"
)
else:
error_message = f"This example requires a minimum version of {min_version},"
error_message += f" but the version found is {__version__}.\n"
raise ImportError(error_message)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment