File size: 331 Bytes
25ac67d |
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
|