Directory rename (git mv, history preserved) plus rewrite of all references: absolute imports in tests, the zero-dep runner, docstrings, comments, and the README. No behavior change. Run: python3 -m pytest v2/tests/ -q ; python3 v2/run_tests.py ; python3 -m v2.examples
17 lines
646 B
Python
17 lines
646 B
Python
"""Training methods — all on Wan2.1-1.3B, all over the SHARED loops (design_v3 §10)."""
|
|
from __future__ import annotations
|
|
|
|
from .base import TrainingMethod, new_instance, predict_x0
|
|
from .diffusion_nft import DiffusionNFTMethod, build_diffusion_nft
|
|
from .dmd2 import DMD2Method, build_dmd2
|
|
from .finetune import FineTuneMethod, build_finetune
|
|
from .self_forcing import SelfForcingMethod, build_self_forcing
|
|
|
|
__all__ = [
|
|
"TrainingMethod", "new_instance", "predict_x0",
|
|
"FineTuneMethod", "build_finetune",
|
|
"DMD2Method", "build_dmd2",
|
|
"DiffusionNFTMethod", "build_diffusion_nft",
|
|
"SelfForcingMethod", "build_self_forcing",
|
|
]
|