File size: 1,345 Bytes
4689c2b | 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 | # FlashVSR model implementations
from .model_manager import ModelManager
from .wan_video_dit import WanModel, WanModelStateDictConverter
from .wan_video_vae import WanVideoVAE, WanVideoVAEStateDictConverter
from .TCDecoder import TAEW2_1DiffusersWrapper, build_tcdecoder
from .utils import (
hash_state_dict_keys,
load_state_dict,
load_state_dict_from_folder,
Buffer_LQ4x_Proj,
clean_vram,
get_device_list,
FrameStreamBuffer,
TensorAsBuffer,
tensor_to_imageio_frame
)
from .download_manager import (
ensure_models_downloaded,
get_model_paths,
load_pipeline,
check_vram_requirements,
get_available_variants,
estimate_vram_for_resolution
)
__all__ = [
'ModelManager',
'WanModel',
'WanModelStateDictConverter',
'WanVideoVAE',
'WanVideoVAEStateDictConverter',
'TAEW2_1DiffusersWrapper',
'build_tcdecoder',
'hash_state_dict_keys',
'load_state_dict',
'load_state_dict_from_folder',
'Buffer_LQ4x_Proj',
'clean_vram',
'get_device_list',
'FrameStreamBuffer',
'TensorAsBuffer',
'tensor_to_imageio_frame',
'ensure_models_downloaded',
'get_model_paths',
'load_pipeline',
'check_vram_requirements',
'get_available_variants',
'estimate_vram_for_resolution'
]
|