File tree Expand file tree Collapse file tree 5 files changed +906
-0
lines changed Expand file tree Collapse file tree 5 files changed +906
-0
lines changed Original file line number Diff line number Diff line change 462462 "IFPipeline" ,
463463 "IFSuperResolutionPipeline" ,
464464 "ImageTextPipelineOutput" ,
465+ "InfiniteTalkPipeline" ,
465466 "Kandinsky3Img2ImgPipeline" ,
466467 "Kandinsky3Pipeline" ,
467468 "KandinskyCombinedPipeline" ,
11111112 IFPipeline ,
11121113 IFSuperResolutionPipeline ,
11131114 ImageTextPipelineOutput ,
1115+ InfiniteTalkPipeline ,
11141116 Kandinsky3Img2ImgPipeline ,
11151117 Kandinsky3Pipeline ,
11161118 KandinskyCombinedPipeline ,
Original file line number Diff line number Diff line change 240240 "HunyuanVideoImageToVideoPipeline" ,
241241 "HunyuanVideoFramepackPipeline" ,
242242 ]
243+ _import_structure ["infinitetalk" ] = "InfiniteTalkPipeline"
243244 _import_structure ["kandinsky" ] = [
244245 "KandinskyCombinedPipeline" ,
245246 "KandinskyImg2ImgCombinedPipeline" ,
643644 )
644645 from .hunyuandit import HunyuanDiTPipeline
645646 from .i2vgen_xl import I2VGenXLPipeline
647+ from .infinitetalk import InfiniteTalkPipeline
646648 from .kandinsky import (
647649 KandinskyCombinedPipeline ,
648650 KandinskyImg2ImgCombinedPipeline ,
Original file line number Diff line number Diff line change 1+ from typing import TYPE_CHECKING
2+
3+ from ...utils import (
4+ DIFFUSERS_SLOW_IMPORT ,
5+ OptionalDependencyNotAvailable ,
6+ _LazyModule ,
7+ get_objects_from_module ,
8+ is_torch_available ,
9+ is_transformers_available ,
10+ )
11+
12+
13+ _dummy_objects = {}
14+ _import_structure = {"pipeline_output" : ["InfiniteTalkPipelineOutput" ]}
15+
16+
17+ try :
18+ if not (is_transformers_available () and is_torch_available ()):
19+ raise OptionalDependencyNotAvailable ()
20+ except OptionalDependencyNotAvailable :
21+ from ...utils import dummy_torch_and_transformers_objects # noqa F403
22+
23+ _dummy_objects .update (get_objects_from_module (dummy_torch_and_transformers_objects ))
24+ else :
25+ _import_structure ["pipeline_infinitetalk" ] = ["InfiniteTalkPipeline" ]
26+ if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT :
27+ try :
28+ if not (is_transformers_available () and is_torch_available ()):
29+ raise OptionalDependencyNotAvailable ()
30+
31+ except OptionalDependencyNotAvailable :
32+ from ...utils .dummy_torch_and_transformers_objects import *
33+ else :
34+ from .pipeline_infinitetalk import InfiniteTalkPipeline
35+
36+ else :
37+ import sys
38+
39+ sys .modules [__name__ ] = _LazyModule (
40+ __name__ ,
41+ globals ()["__file__" ],
42+ _import_structure ,
43+ module_spec = __spec__ ,
44+ )
45+
46+ for name , value in _dummy_objects .items ():
47+ setattr (sys .modules [__name__ ], name , value )
You can’t perform that action at this time.
0 commit comments