Spaces:
Running
on
Zero
Running
on
Zero
File size: 3,268 Bytes
d2c9b66 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 |
# from .pipeline_cogvideox_fun import CogVideoXFunPipeline
# from .pipeline_cogvideox_fun_control import CogVideoXFunControlPipeline
# from .pipeline_cogvideox_fun_inpaint import CogVideoXFunInpaintPipeline
# from .pipeline_fantasy_talking import FantasyTalkingPipeline
# from .pipeline_flux import FluxPipeline
# from .pipeline_flux2 import Flux2Pipeline
# from .pipeline_flux2_control import Flux2ControlPipeline
# from .pipeline_hunyuanvideo import HunyuanVideoPipeline
# from .pipeline_hunyuanvideo_i2v import HunyuanVideoI2VPipeline
# from .pipeline_qwenimage import QwenImagePipeline
# from .pipeline_qwenimage_edit import QwenImageEditPipeline
# from .pipeline_qwenimage_edit_plus import QwenImageEditPlusPipeline
# from .pipeline_wan import WanPipeline
# from .pipeline_wan2_2 import Wan2_2Pipeline
# from .pipeline_wan2_2_animate import Wan2_2AnimatePipeline
# from .pipeline_wan2_2_fun_control import Wan2_2FunControlPipeline
# from .pipeline_wan2_2_fun_inpaint import Wan2_2FunInpaintPipeline
# from .pipeline_wan2_2_s2v import Wan2_2S2VPipeline
# from .pipeline_wan2_2_ti2v import Wan2_2TI2VPipeline
# from .pipeline_wan2_2_vace_fun import Wan2_2VaceFunPipeline
# from .pipeline_wan_fun_control import WanFunControlPipeline
# from .pipeline_wan_fun_inpaint import WanFunInpaintPipeline
# from .pipeline_wan_phantom import WanFunPhantomPipeline
# from .pipeline_wan_vace import WanVacePipeline
from .pipeline_z_image import ZImagePipeline
from .pipeline_z_image_control import ZImageControlPipeline
# WanFunPipeline = WanPipeline
# WanI2VPipeline = WanFunInpaintPipeline
# Wan2_2FunPipeline = Wan2_2Pipeline
# Wan2_2I2VPipeline = Wan2_2FunInpaintPipeline
# import importlib.util
# if importlib.util.find_spec("paifuser") is not None:
# # --------------------------------------------------------------- #
# # Sparse Attention
# # --------------------------------------------------------------- #
# from paifuser.ops import sparse_reset
# # Wan2.1
# WanFunInpaintPipeline.__call__ = sparse_reset(WanFunInpaintPipeline.__call__)
# WanFunPipeline.__call__ = sparse_reset(WanFunPipeline.__call__)
# WanFunControlPipeline.__call__ = sparse_reset(WanFunControlPipeline.__call__)
# WanI2VPipeline.__call__ = sparse_reset(WanI2VPipeline.__call__)
# WanPipeline.__call__ = sparse_reset(WanPipeline.__call__)
# WanVacePipeline.__call__ = sparse_reset(WanVacePipeline.__call__)
# # Phantom
# WanFunPhantomPipeline.__call__ = sparse_reset(WanFunPhantomPipeline.__call__)
# # Wan2.2
# Wan2_2FunInpaintPipeline.__call__ = sparse_reset(Wan2_2FunInpaintPipeline.__call__)
# Wan2_2FunPipeline.__call__ = sparse_reset(Wan2_2FunPipeline.__call__)
# Wan2_2FunControlPipeline.__call__ = sparse_reset(Wan2_2FunControlPipeline.__call__)
# Wan2_2Pipeline.__call__ = sparse_reset(Wan2_2Pipeline.__call__)
# Wan2_2I2VPipeline.__call__ = sparse_reset(Wan2_2I2VPipeline.__call__)
# Wan2_2TI2VPipeline.__call__ = sparse_reset(Wan2_2TI2VPipeline.__call__)
# Wan2_2S2VPipeline.__call__ = sparse_reset(Wan2_2S2VPipeline.__call__)
# Wan2_2VaceFunPipeline.__call__ = sparse_reset(Wan2_2VaceFunPipeline.__call__)
# Wan2_2AnimatePipeline.__call__ = sparse_reset(Wan2_2AnimatePipeline.__call__) |