ColossalAI/colossalai
Xu Kai 20332a7a34
[inference] udpate example (#5053)
* udpate example

* fix run.sh
2023-11-16 11:07:43 +08:00
..
_C [setup] support pre-build and jit-build of cuda kernels (#2374) 2023-01-06 20:50:26 +08:00
_analyzer [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
amp [feature] Add clip_grad_norm for hybrid_parallel_plugin (#4837) 2023-10-12 11:32:37 +08:00
auto_parallel [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
autochunk [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
booster [moe] support optimizer checkpoint (#5015) 2023-11-08 15:07:03 +00:00
checkpoint_io [checkpointio] hotfix torch 2.0 compatibility (#4824) 2023-10-07 10:45:52 +08:00
cli [bug] Fix the version check bug in colossalai run when generating the cmd. (#4713) 2023-09-22 10:50:47 +08:00
cluster [hotfix] Fix the bug where process groups were not being properly released. (#4940) 2023-10-31 14:47:30 +08:00
context [moe] merge moe into main (#4978) 2023-11-02 02:21:24 +00:00
device [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
fx [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
inference [inference] udpate example (#5053) 2023-11-16 11:07:43 +08:00
interface [lazy] support from_pretrained (#4801) 2023-09-26 11:04:11 +08:00
kernel [moe] merge moe into main (#4978) 2023-11-02 02:21:24 +00:00
lazy [doc] add lazy init docs (#4808) 2023-09-27 10:24:04 +08:00
legacy [inference] udpate example (#5053) 2023-11-16 11:07:43 +08:00
logging [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
moe [moe] support optimizer checkpoint (#5015) 2023-11-08 15:07:03 +00:00
nn [moe] merge moe into main (#4978) 2023-11-02 02:21:24 +00:00
pipeline [Refactor] remove useless inference code (#5022) 2023-11-10 14:47:06 +08:00
shardformer [Refactor] refactor policy search and quant type controlling in inference (#5035) 2023-11-14 17:26:59 +08:00
tensor [moe] support optimizer checkpoint (#5015) 2023-11-08 15:07:03 +00:00
testing [test] merge old components to test to model zoo (#4945) 2023-10-20 10:35:08 +08:00
utils [moe] merge moe into main (#4978) 2023-11-02 02:21:24 +00:00
zero [hotfix] Add layer norm gradients all-reduce for sequence parallel (#4926) 2023-11-03 13:32:43 +08:00
__init__.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
initialize.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00