You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
6 lines
404 B
Python
6 lines
404 B
Python
from src.client.inference_session import RemoteSequentialInferenceSession, RemoteTransformerBlockInferenceSession
|
|
from src.client.remote_block import RemoteTransformerBlock
|
|
from src.client.remote_model import DistributedBloomConfig, DistributedBloomForCausalLM, DistributedBloomModel
|
|
from src.client.remote_sequential import RemoteSequential
|
|
from src.client.sequence_manager import RemoteSequenceManager
|