feat: migrate switch to conditional flow matching from sphere trajectory

2026-01-22 14:37:50 +08:00
parent c15115edc4
commit 444f5fc109
5 changed files with 1264 additions and 455 deletions
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -5,6 +5,7 @@ description = "Add your description here"
 readme = "README.md"
 requires-python = ">=3.12"
 dependencies = [
+    "datasets>=2.19.0",
    "einops>=0.7.0",
    "matplotlib>=3.8.0",
    "numpy>=1.26.0",