Frank Lee
|
c4b1b65931
|
[test] fixed tests failed due to dtensor change (#4082)
* [test] fixed tests failed due to dtensor change
* polish code
|
2023-07-04 16:05:01 +08:00 |
|
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
|
2023-06-15 17:38:42 +08:00 |
|
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
|
2023-04-06 14:51:35 +08:00 |
|
Xuanlei Zhao
|
10c61de2f7
|
[autochunk] support vit (#3084)
support vit for autochunk
* support some new ops for vit
* fix some bugs
* add test for vit
|
2023-03-10 10:23:26 +08:00 |
|
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
|
2023-02-07 16:32:45 +08:00 |
|
oahzxl
|
c4b15661d7
|
[autochunk] add benchmark for transformer and alphafold (#2543)
|
2023-02-02 15:06:43 +08:00 |