ColossalAI

Commit Graph

Author	SHA1	Message	Date
Jiarui Fang	6fa71d65d3	[fx] skip diffusers unitest if it is not installed (#1799 )	2022-11-08 11:45:23 +08:00
Super Daniel	5ea89f6456	[CI] downgrade fbgemm. (#1778 )	2022-10-31 18:18:45 +08:00
oahzxl	25952b67d7	[feat] add flash attention (#1762 )	2022-10-26 16:15:52 +08:00
Super Daniel	b893342f95	[fx] test tracer on diffuser modules. (#1750 ) * [fx] test tracer on diffuser modules. * [fx] shorter seq_len. * Update requirements-test.txt	2022-10-20 18:25:05 +08:00
Jiarui Fang	504419d261	[FAW] add cache manager for the cached embedding (#1419 )	2022-08-09 15:17:17 +08:00
Super Daniel	be229217ce	[fx] add torchaudio test (#1369 ) * [fx]add torchaudio test * [fx]add torchaudio test * [fx] add torchaudio test * [fx] add torchaudio test * [fx] add torchaudio test * [fx] add torchaudio test * [fx] add torchaudio test * [fx] add torchaudio test and test patches * Delete ~ * [fx] add patches and patches test * [fx] add patches and patches test * [fx] fix patches * [fx] fix rnn patches * [fx] fix rnn patches * [fx] fix rnn patches * [fx] fix rnn patches * [fx] merge upstream * [fx] fix import errors	2022-07-27 11:03:14 +08:00
Boyuan Yao	bb640ec728	[fx] Add colotracer compatibility test on torchrec (#1370 )	2022-07-26 17:54:39 +08:00
Frank Lee	b2475d8c5c	[fx] fixed unit tests for torch 1.12 (#1327 )	2022-07-15 18:22:15 +08:00
YuliangLiu0306	9feff0f760	[titans]remove model zoo (#1042 ) * [CLI] add CLI launcher * Revert "[CLI] add CLI launcher" This reverts commit `df7e6506d4`. * rm model zoo	2022-05-31 10:40:47 +08:00
Frank Lee	01e9f834f5	[dependency] removed torchvision (#833 ) * [dependency] removed torchvision * fixed transforms	2022-04-22 15:24:35 +08:00
ver217	70e8dd418b	[hotfix] update requirements-test (#701 )	2022-04-08 16:52:36 +08:00
Jiarui Fang	54229cd33e	[log] better logging display with rich (#426 ) * better logger using rich * remove deepspeed in zero requirements	2022-03-16 09:51:15 +08:00
zbian	404ecbdcc6	Migrated project	2021-10-28 18:21:23 +02:00

13 Commits (7080a8edb08400d97ba4c31458f532e4ceeacf4b)