hpcaitech/ColossalAI-Examples

ZeRO 2 configuration example

CHN-ChenYi opened this issue ยท 1 comments

๐Ÿ“š The doc issue

The gpt2 zero2 example was removed after the new API was introduced.
But I don't know how to not offload the model.
I've tried removing model_config or setting offload_config to None/gpu/cuda, but neither of them works.