SolitaryThinker
db866dfb04
Sweep all 296 v2 modules: simplify verbose docstrings/comments and remove 376 dangling "(design_vN §X)" citations to the now-deleted design docs (v2/README.md is the source of truth). Comment/docstring-only — AST-verified code-identical; the CPU suite holds at 237 passed / 5 pre-existing. Also applies yapf + ruff --fix auto-fixes (import ordering, forward-ref annotation de-quoting under `from __future__ import annotations`; behavior- neutral, suite-confirmed) and adds the legitimate domain terms mot/clen/te to the codespell ignore-list. Remaining ruff (24) + mypy (68 no-untyped-call) findings are pre-existing v2 debt, untouched here.
32 lines
1.1 KiB
Python
32 lines
1.1 KiB
Python
"""Training methods — all on Wan2.1-1.3B, all over the SHARED loops."""
|
|
from __future__ import annotations
|
|
|
|
from v2.training.methods.base import TrainingMethod, new_instance, predict_x0
|
|
from v2.training.methods.diffusion_nft import DiffusionNFTMethod, build_diffusion_nft
|
|
from v2.training.methods.dmd2 import DMD2Method, build_dmd2
|
|
from v2.training.methods.finetune import FineTuneMethod, build_finetune
|
|
from v2.training.methods.joint_multi_rl import JointMultiExpertRL, build_joint_multi_rl
|
|
from v2.training.methods.self_forcing import SelfForcingMethod, build_self_forcing
|
|
from v2.training.methods.unified_rl import UnifiedRLMethod, build_unified_rl
|
|
from v2.training.methods.workflow_rl import WorkflowRLMethod, build_workflow_rl
|
|
|
|
__all__ = [
|
|
"TrainingMethod",
|
|
"new_instance",
|
|
"predict_x0",
|
|
"FineTuneMethod",
|
|
"build_finetune",
|
|
"DMD2Method",
|
|
"build_dmd2",
|
|
"DiffusionNFTMethod",
|
|
"build_diffusion_nft",
|
|
"SelfForcingMethod",
|
|
"build_self_forcing",
|
|
"UnifiedRLMethod",
|
|
"build_unified_rl",
|
|
"JointMultiExpertRL",
|
|
"build_joint_multi_rl",
|
|
"WorkflowRLMethod",
|
|
"build_workflow_rl",
|
|
]
|