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.
petals/src
Alexander Borzunov 8a73b41a42
Make ServerState announcements work better (#93)
- Before this PR, `ServerState.JOINING` was announced only once. This announcement quickly expires in case of the full-size BLOOM, since loading blocks takes several minutes. This PR fixes it, so `ServerState.JOINING` is announced periodically in a thread until blocks are loaded.

- This PR also makes the `Server` class a non-thread, so it runs in the main thread and can catch `KeyboardInterrupt`. This is important, since if we are downloading blocks right now, we need to stop it and send the `ServerState.OFFLINE` message. Note that `ModuleContainer` is still a thread.

- (minor) For the sake of readability, I moved the `ModuleContainer.create()` definition, so it is now defined before `Server.__init__()` (this is because `.create()` is invoked first).
2 years ago
..
bloom [Fix] make distributed seq cls to not create the full bloom model (#49) 2 years ago
client Use public swarm by default (#92) 2 years ago
server Make ServerState announcements work better (#93) 2 years ago
utils Use bitsandbytes==0.34.0, update readme (#76) 2 years ago
__init__.py Measure and cache network & compute throughput (#21) 2 years ago
constants.py Use public swarm by default (#92) 2 years ago
data_structures.py Implement RemoteSequential slicing and extra repr, add tests (#30) 2 years ago
dht_utils.py remove transformer block, implement as sequential of size 1 (#54) 2 years ago