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/petals
Alexander Borzunov 26ebbfe8f0
Support macOS (#477)
This PR makes both clients and servers work on macOS. Specifically, it:

- Follows https://github.com/learning-at-home/hivemind/pull/586 to run a macOS-compatible `p2pd` binary (both x86-64 and ARM64 are supported)
- Fixes forking issues and tests on macOS, Python 3.10+
- Introduces basic support for serving model blocks on Apple M1/M2 GPUs (torch.mps)
- Increases max number of open files by default (it's not enough on Linux and is really small on macOS)
9 months ago
..
cli Support macOS (#477) 9 months ago
client Make client compatible with transformers' GenerationMixin (#464) 10 months ago
models Make client compatible with transformers' GenerationMixin (#464) 10 months ago
server Support macOS (#477) 9 months ago
utils Rewrite MemoryCache alloc_timeout logic (#434) 9 months ago
__init__.py Support macOS (#477) 9 months ago
constants.py Update to petals.dev (#390) 11 months ago
data_structures.py Move SequenceManagerConfig -> ClientConfig, petals.dht_utils -> petals.utils.dht (#463) 10 months ago
dht_utils.py Move SequenceManagerConfig -> ClientConfig, petals.dht_utils -> petals.utils.dht (#463) 10 months ago