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 dd4a3230bc
Add Falcon support (#499)
This PR adds:

- Support for models based on `transformers.FalconModel` (the in-library format for Falcon). Tested on Falcon-40B.
- CI tests for Falcon-RW-1B.
- `--throughput dry_run` option to evaluate throughput and exit right away (implemented by @mryab).

Limitations:

- Backward pass support is broken for now, will be fixed in #500.

Co-authored-by: Max Ryabinin <mryabinin0@gmail.com>
9 months ago
..
cli Add Falcon support (#499) 9 months ago
client Create model index in DHT (#491) 9 months ago
models Add Falcon support (#499) 9 months ago
server Add Falcon support (#499) 9 months ago
utils Add Falcon support (#499) 9 months ago
__init__.py Support macOS (#477) 10 months ago
constants.py Update to petals.dev (#390) 11 months ago
data_structures.py Create model index in DHT (#491) 9 months ago
dht_utils.py Move SequenceManagerConfig -> ClientConfig, petals.dht_utils -> petals.utils.dht (#463) 10 months ago