Xu Kai 785802e809 [inference] add reference and fix some bugs (#4937) ...

* add reference and fix some bugs

* update gptq init

---------

Co-authored-by: Xu Kai <xukai16@foxamil.com>

2023-10-20 13:39:34 +08:00

..

_analyzer

[misc] update pre-commit and run all files (#4752)

2023-09-19 14:20:26 +08:00

_C

[setup] support pre-build and jit-build of cuda kernels (#2374)

2023-01-06 20:50: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

[gemini] support gradient accumulation (#4869)

2023-10-17 14:07:21 +08: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

[doc] polish shardformer doc (#4779)

2023-09-26 10:57:47 +08:00

context

[misc] update pre-commit and run all files (#4752)

2023-09-19 14:20:26 +08: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] add reference and fix some bugs (#4937)

2023-10-20 13:39:34 +08:00

interface

[lazy] support from_pretrained (#4801)

2023-09-26 11:04:11 +08:00

kernel

[inference] add reference and fix some bugs (#4937)

2023-10-20 13:39:34 +08:00

lazy

[doc] add lazy init docs (#4808)

2023-09-27 10:24:04 +08:00

legacy

[hotfix] fix torch 2.0 compatibility (#4936)

2023-10-18 11:05:25 +08:00

logging

[misc] update pre-commit and run all files (#4752)

2023-09-19 14:20:26 +08:00

nn

[test] add no master test for low level zero plugin (#4934)

2023-10-18 11:41:23 +08:00

pipeline

[format] applied code formatting on changed files in pull request 4820 (#4886)

2023-10-18 11:46:37 +08:00

shardformer

[nfc] fix some typo with colossalai/ docs/ etc. (#4920)

2023-10-18 15:44:04 +08:00

tensor

[misc] update pre-commit and run all files (#4752)

2023-09-19 14:20:26 +08:00

testing

[test] merge old components to test to model zoo (#4945)

2023-10-20 10:35:08 +08:00

utils

[misc] update pre-commit and run all files (#4752)

2023-09-19 14:20:26 +08:00

zero

[gemini] support gradient accumulation (#4869)

2023-10-17 14:07:21 +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