|
@@ -10,7 +10,7 @@ from .layers import name_to_block
|
|
|
|
|
|
def make_dummy_server(host='0.0.0.0', port=None, num_experts=1, expert_cls='ffn', hidden_dim=1024, num_handlers=None,
|
|
def make_dummy_server(host='0.0.0.0', port=None, num_experts=1, expert_cls='ffn', hidden_dim=1024, num_handlers=None,
|
|
expert_prefix='expert.', expert_offset=0, max_batch_size=16384, device=None, no_optimizer=False,
|
|
expert_prefix='expert.', expert_offset=0, max_batch_size=16384, device=None, no_optimizer=False,
|
|
- no_network=False, initial_peers=(), network_port=None, root_port=None, verbose=True, start=True,
|
|
|
|
|
|
+ no_network=False, initial_peers=(), network_port=None, root_port=None, verbose=True, start=False,
|
|
**kwargs) -> tesseract.TesseractServer:
|
|
**kwargs) -> tesseract.TesseractServer:
|
|
""" A context manager that creates server in a background thread, awaits .ready on entry and shutdowns on exit """
|
|
""" A context manager that creates server in a background thread, awaits .ready on entry and shutdowns on exit """
|
|
if verbose and len(kwargs) != 0:
|
|
if verbose and len(kwargs) != 0:
|