Hongxin Liu
|
079bf3cb26
|
[misc] update pre-commit and run all files (#4752)
* [misc] update pre-commit
* [misc] run pre-commit
* [misc] remove useless configuration files
* [misc] ignore cuda for clang-format
|
1 year ago |
|
Frank Lee
|
c4b1b65931
|
[test] fixed tests failed due to dtensor change (#4082)
* [test] fixed tests failed due to dtensor change
* polish code
|
1 year ago |
|
Wenhao Chen
|
725af3eeeb
|
[booster] make optimizer argument optional for boost (#3993)
* feat: make optimizer optional in Booster.boost
* test: skip unet test if diffusers version > 0.10.2
|
1 year ago |
|
Frank Lee
|
80eba05b0a
|
[test] refactor tests with spawn (#3452)
* [test] added spawn decorator
* polish code
* polish code
* polish code
* polish code
* polish code
* polish code
|
2 years ago |
|
Xuanlei Zhao
|
10c61de2f7
|
[autochunk] support vit (#3084)
support vit for autochunk
* support some new ops for vit
* fix some bugs
* add test for vit
|
2 years ago |
|
oahzxl
|
6ba8364881
|
[autochunk] support diffusion for autochunk (#2621)
* add alphafold benchmark
* renae alphafold test
* rename tests
* rename diffuser
* renme
* rename
* update transformer
* update benchmark
* update benchmark
* update bench memory
* update transformer benchmark
* rename
* support diffuser
* support unet metainfo prop
* fix bug and simplify code
* update linear and support some op
* optimize max region search, support conv
* update unet test
* support some op
* support groupnorm and interpolate
* update flow search
* add fix dim in node flow
* fix utils
* rename
* support diffusion
* update diffuser
* update chunk search
* optimize imports
* import
* finish autochunk
|
2 years ago |
|
oahzxl
|
c4b15661d7
|
[autochunk] add benchmark for transformer and alphafold (#2543)
|
2 years ago |
|
oahzxl
|
63199c6687
|
[autochunk] support transformer (#2526)
|
2 years ago |
