ColossalAI/requirements.txt at feature/elixir - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-08-29 13:13:44 +00:00

Zihao 18dbe76cae

[auto-parallel] add auto-offload feature (#3154 )

* add auto-offload feature

* polish code

* fix syn offload runtime pass bug

* add offload example

* fix offload testing bug

* fix example testing bug

2023-03-21 14:17:41 +08:00

2 lines

35 B

Plaintext

Raw Permalink Blame History

	`colossalai >= 0.1.12`
	`torch >= 1.8.1`