fix(pusht): stabilize DiT pusht training on current stack

2026-03-15 18:54:50 +08:00
parent 08c1950c6d
commit 2aa06c8917
2 changed files with 3 additions and 3 deletions
--- a/requirements-pusht-5090.txt
+++ b/requirements-pusht-5090.txt
@@ -22,6 +22,7 @@ pymunk==6.2.1
 wandb==0.13.3
 threadpoolctl==3.1.0
 shapely==1.8.5.post1
+matplotlib==3.6.1
 imageio==2.22.0
 imageio-ffmpeg==0.4.7
 termcolor==2.0.1