This commit is contained in:
@@ -0,0 +1,7 @@
|
||||
# Copyright (c) ModelScope Contributors. All rights reserved.
|
||||
import os
|
||||
|
||||
if __name__ == '__main__':
|
||||
os.environ.setdefault('CUDA_DEVICE_MAX_CONNECTIONS', '1')
|
||||
from swift.megatron import megatron_export_main
|
||||
megatron_export_main()
|
||||
Reference in New Issue
Block a user