mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-04-29 04:05:35 +00:00
50793b35f4
* [gemini] support don't scatter after inference * [chat] update colossalai strategy * [chat] fix opt benchmark * [chat] update opt benchmark * [gemini] optimize inference * [test] add gemini inference test * [chat] fix unit test ci * [chat] fix ci * [chat] fix ci * [chat] skip checkpoint test
3 lines
96 B
Python
3 lines
96 B
Python
from .dummy_data_generator import DummyDataGenerator
|
|
from .executor import run_fwd, run_fwd_bwd
|