hongshaorou/dreamer4
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

bring in muon

Browse Source
This commit is contained in:
lucidrains 2025-10-12 09:35:06 -07:00
parent 8a73a27fc7
commit ab5de6795f
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
pyproject.toml
View File

@ -27,6 +27,7 @@ classifiers=[
dependencies = [ dependencies = [
"accelerate", "accelerate",
"adam-atan2-pytorch>=0.2.2",
"assoc-scan", "assoc-scan",
"einx>=0.3.0", "einx>=0.3.0",
"einops>=0.8.1", "einops>=0.8.1",