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

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-08-30 23:56:42 +00:00

Files

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`