Implement Mamba MeanFlow x-prediction training

2026-03-11 16:33:40 +08:00
parent 01fc1e4eab
commit 9b2968997c
5 changed files with 353 additions and 121 deletions
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -7,8 +7,10 @@ requires-python = ">=3.12"
 dependencies = [
    "datasets>=2.19.0",
    "einops>=0.7.0",
+    "lpips>=0.1.4",
    "matplotlib>=3.8.0",
    "numpy>=1.26.0",
    "swanlab>=0.5.0",
    "torch>=2.2.0",
+    "torchvision>=0.24.1",
 ]