|
|
|
@ -14,7 +14,6 @@ logger = get_logger(__name__)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@pytest.mark.forked
|
|
|
|
|
@pytest.mark.skip(reason="no way of currently testing this")
|
|
|
|
|
def test_remote_sequential():
|
|
|
|
|
config = AutoDistributedConfig.from_pretrained(MODEL_NAME, initial_peers=INITIAL_PEERS)
|
|
|
|
|
dht = DHT(initial_peers=config.initial_peers, client_mode=True, start=True)
|
|
|
|
@ -87,7 +86,6 @@ class DummyCustomSequenceManager(RemoteSequenceManager):
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@pytest.mark.forked
|
|
|
|
|
@pytest.mark.skip(reason="no way of currently testing this")
|
|
|
|
|
def test_remote_sequential_prompts(batch_size=2, seq_len=5, pre_seq_len=3):
|
|
|
|
|
config = AutoDistributedConfig.from_pretrained(MODEL_NAME, initial_peers=INITIAL_PEERS)
|
|
|
|
|
remote_sequential = RemoteSequential(config)
|
|
|
|
|