File tree Expand file tree Collapse file tree 4 files changed +16
-15
lines changed Expand file tree Collapse file tree 4 files changed +16
-15
lines changed Original file line number Diff line number Diff line change 1818)
1919
2020
21- # Make sure `transformers` is up to date
22- if is_transformers_available ():
23- import transformers
24-
25- if is_transformers_version ("<" , "4.25.1" ):
26- raise ImportError (
27- f"`diffusers` requires transformers >= 4.25.1 to function correctly, but { transformers .__version__ } was"
28- " found in your environment. You can upgrade it with pip: `pip install transformers --upgrade`"
29- )
30- else :
31- pass
32-
3321try :
3422 if not is_torch_available ():
3523 raise OptionalDependencyNotAvailable ()
Original file line number Diff line number Diff line change @@ -46,7 +46,7 @@ class StableDiffusionPipelineOutput(BaseOutput):
4646 from .safety_checker import StableDiffusionSafetyChecker
4747
4848try :
49- if not (is_transformers_available () and is_torch_available ()):
49+ if not (is_transformers_available () and is_torch_available () and is_transformers_version ( ">=" , "4.25.0" ) ):
5050 raise OptionalDependencyNotAvailable ()
5151except OptionalDependencyNotAvailable :
5252 from ...utils .dummy_torch_and_transformers_objects import StableDiffusionImageVariationPipeline
Original file line number Diff line number Diff line change 77
88
99try :
10- if not (is_transformers_available () and is_torch_available ()):
10+ if not (is_transformers_available () and is_torch_available () and is_transformers_version ( ">=" , "4.25.0" ) ):
1111 raise OptionalDependencyNotAvailable ()
1212except OptionalDependencyNotAvailable :
1313 from ...utils .dummy_torch_and_transformers_objects import (
Original file line number Diff line number Diff line change @@ -354,7 +354,20 @@ def requires_backends(obj, backends):
354354 if failed :
355355 raise ImportError ("" .join (failed ))
356356
357- if name in ["StableDiffusionDepth2ImgPipeline" ] and is_transformers_version ("<" , "4.26.0.dev0" ):
357+ if name in [
358+ "VersatileDiffusionTextToImagePipeline" ,
359+ "VersatileDiffusionPipeline" ,
360+ "VersatileDiffusionDualGuidedPipeline" ,
361+ "StableDiffusionImageVariationPipeline" ,
362+ ] and is_transformers_version ("<" , "4.25.0" ):
363+ raise ImportError (
364+ f"You need to install `transformers>=4.25` in order to use { name } : \n ```\n pip install"
365+ " --upgrade transformers \n ```"
366+ )
367+
368+ if name in [
369+ "StableDiffusionDepth2ImgPipeline" ,
370+ ] and is_transformers_version ("<" , "4.26.0.dev0" ):
358371 raise ImportError (
359372 f"You need to install `transformers` from 'main' in order to use { name } : \n ```\n pip install"
360373 " git+https://github.com/huggingface/transformers \n ```"
You can’t perform that action at this time.
0 commit comments