Add diffusers version check

This commit is contained in:
huchenlei
2024-03-07 23:24:21 -05:00
parent 9b55ebbd13
commit 2767a5b0f4
+12
View File
@@ -10,6 +10,18 @@ from diffusers.models.modeling_utils import ModelMixin
from diffusers.models.unet_2d_blocks import UNetMidBlock2D, get_down_block, get_up_block
def check_diffusers_version():
import diffusers
from packaging.version import parse
assert parse(diffusers.__version__) >= parse(
"0.25.0"
), "diffusers>=0.25.0 requirement not satisfied. Please install correct diffusers version."
check_diffusers_version()
def zero_module(module):
"""
Zero out the parameters of a module and return it.