Duplicated from akhaliq/lama
ff8c072
1
2
from .data_parallel import UserScatteredDataParallel, user_scattered_collate, async_copy_to
