Spaces:
Running on Zero
Running on Zero
File size: 185 Bytes
ee69ab7 | 1 2 3 4 5 6 7 8 | from .config import FlashVSRModelConfig
from .dit import FlashVSRDiT
def build_model(config: FlashVSRModelConfig) -> FlashVSRDiT:
config.validate()
return FlashVSRDiT(config)
|