16 lines
674 B
Python
16 lines
674 B
Python
"""BasicVSR++ generator (vendored, mmengine-free).
|
|
|
|
This package re-implements the BasicVSR++ generator network and its SPyNet
|
|
dependency in plain PyTorch so that ComfyUI-MosaicRestore can load lada's
|
|
``lada_mosaic_restoration_model_generic_v1.x.pth`` weights without pulling in
|
|
``mmengine``/``mmagic``.
|
|
|
|
State dict compatibility: the vendored module names mirror the keys produced
|
|
by lada's training script, so the lada checkpoint loads with ``strict=True``
|
|
once the EMA prefix is stripped (see :func:`load_basicvsrpp_checkpoint`).
|
|
"""
|
|
|
|
from .net import BasicVSRPlusPlusGanNet, load_basicvsrpp_checkpoint
|
|
|
|
__all__ = ["BasicVSRPlusPlusGanNet", "load_basicvsrpp_checkpoint"]
|