from src.client.remote_block import RemoteTransformerBlock, RemoteTransformerBlockInferenceSession