File size: 331 Bytes
5bdad4f
 
 
 
1
2
3
4
5
from src.client.remote_block import RemoteTransformerBlock, RemoteTransformerBlockInferenceSession
from src.client.remote_model import DistributedBloomConfig, DistributedBloomForCausalLM, DistributedBloomModel
from src.client.remote_sequence_info import RemoteSequenceInfo
from src.client.remote_sequential import RemoteSequential