Student0809's picture
Add files using upload-large-folder tool
7feac49 verified
raw
history blame
216 Bytes
import os
if os.environ.get('SEQUENCE_PARALLEL_IMPL', 'ulysses') == 'xtuner':
from .xtuner import XTuner
sequence_parallel = XTuner()
else:
from .ulysses import Ulysses
sequence_parallel = Ulysses()