Hugging Face's logo

Spaces:
hugging-apps
/
flashvsr-plus
Running on Zero

App Files Community
flashvsr-plus
File size: 1,210 Bytes
ee69ab7
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
from .attention import AttentionRuntime, build_block_causal_mask
from .attention_backends import attention_kernel_status
from .config import (
    AttentionConfig,
    BlockReplacementConfig,
    CausalAttentionConfig,
    DiTConfig,
    FlashVSRModelConfig,
    SlidingAttentionConfig,
    LQProjectionConfig,
    SparseAttentionConfig,
)
from .conditioning import LQProjection
from .dit import FlashVSRDiT
from .factory import build_model
from .vae import (
    WanVAE,
    WanVAEEncoder,
    build_vae,
    build_vae_encoder,
    build_wan_vae,
    build_wan_vae_encoder,
    load_wan_vae_checkpoint,
    load_wan_vae_encoder_checkpoint,
)

__all__ = [
    "AttentionConfig",
    "AttentionRuntime",
    "BlockReplacementConfig",
    "CausalAttentionConfig",
    "DiTConfig",
    "FlashVSRDiT",
    "FlashVSRModelConfig",
    "LQProjection",
    "LQProjectionConfig",
    "SlidingAttentionConfig",
    "SparseAttentionConfig",
    "build_block_causal_mask",
    "build_model",
    "build_vae",
    "build_vae_encoder",
    "build_wan_vae",
    "build_wan_vae_encoder",
    "load_wan_vae_checkpoint",
    "load_wan_vae_encoder_checkpoint",
    "attention_kernel_status",
    "WanVAE",
    "WanVAEEncoder",
]