mirror of
https://github.com/hpcaitech/Open-Sora.git
synced 2026-05-21 11:59:01 +02:00
18 lines
275 B
Python
18 lines
275 B
Python
|
plugin = "hybrid"
|
||
|
|
plugin_config = dict(
|
||
|
|
tp_size=8,
|
||
|
|
pp_size=1,
|
||
|
|
sp_size=1,
|
||
|
|
zero_stage=2,
|
||
|
|
overlap_allgather=False,
|
||
|
|
)
|
||
|
|
|
||
|
|
plugin_ae = "hybrid"
|
||
|
|
plugin_config_ae = dict(
|
||
|
|
tp_size=8,
|
||
|
|
pp_size=1,
|
||
|
|
sp_size=1,
|
||
|
|
zero_stage=2,
|
||
|
|
overlap_allgather=False,
|
||
|
|
)
|