/
githubmirror
/
ColossalAI
Обзор
Документация
Войти
/
githubmirror
/
ColossalAI
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
main
colossalai/legacy/nn/layer/utils/__init__.py
21 строка
445 B
Hongxin Liu
[misc] update pre-commit and run all files (#4752)
19 сен 2023, 09:20
Не верифицирован
19 сен 2023, 09:20
079bf3c
Код
Авторство
О чём код?
from .common import ( ACT2FN, CheckpointModule, _ntuple, divide, get_tensor_parallel_mode, set_tensor_parallel_attribute_by_partition, set_tensor_parallel_attribute_by_size, to_2tuple, ) __all__ = [ "CheckpointModule", "divide", "ACT2FN", "set_tensor_parallel_attribute_by_size", "set_tensor_parallel_attribute_by_partition", "get_tensor_parallel_mode", "_ntuple", "to_2tuple", ]