Commit Graph

68 Commits

Author SHA1 Message Date
Alexander Borzunov
523a7cad33
Fix issues related to petals as a module (#159)
1. Added `from petals.client import *` to `petals/__init__.py`, so you can write just that:

    ```python
    from petals import DistributedBloomForCausalLM
    ```

    I didn't do the same with server, since its classes are supposed to by used by `petals.cli.run_server`, not end-users. Though it's still possible to do `from petals.server.smth import smth` if necessary.

2. Fixed one more logging issue: log lines from hivemind were shown twice due to a bug in #156.

3. Removed unused `runtime.py`, since the server actually uses `hivemind.moe.Runtime`, and `runtime.py` has no significant changes comparing to it.
2022-12-16 09:09:06 +04:00
Aleksandr Borzunov
f42e559c77 Update README.md 2022-12-09 17:54:00 +00:00
Alexander Borzunov
6beb686909
Add link to privacy & security Wiki (#144) 2022-12-09 21:40:29 +04:00
Alexander Borzunov
e99bf36647
Use common folder for all caches, make it a volume in Dockerfile (#141) 2022-12-09 03:54:57 +04:00
Alexander Borzunov
5f50ea9c79
Update Anaconda instructions (#140) 2022-12-09 01:16:33 +04:00
Alexander Borzunov
4cb0ac4718
Update texts in "Terms of use" and "Privacy and security" sections (#138) 2022-12-08 04:41:02 +04:00
Alexander Borzunov
a94c91d870
Add Docker commands, use permanent Discord links (#137) 2022-12-08 01:54:22 +04:00
Alexander Borzunov
66f1799d32
Set default --step_timeout to 5 min (#133) 2022-12-05 13:44:18 +04:00
Alexander Borzunov
b873d92ffa
Update README.md 2022-12-04 22:48:51 +04:00
Alexander Borzunov
5d5d2666b8
Mention parallel inference 2022-12-04 22:48:32 +04:00
Alexander Borzunov
955eae30b3
Mention 1 sec/token explicitly 2022-12-04 22:10:15 +04:00
Alexander Borzunov
33c210b973
Update Colab notebook 2022-12-03 20:38:18 +04:00
Max Ryabinin
3ca8b4f082
Fix typos with codespell (#126) 2022-12-03 14:19:37 +03:00
Alexander Borzunov
d29ef70c85
Update README.md 2022-12-03 01:14:02 +04:00
Alexander Borzunov
1d9aa77697
Update README.md 2022-12-03 00:46:52 +04:00
Alexander Borzunov
da36470a4b
Update README.md 2022-12-03 00:46:08 +04:00
Alexander Borzunov
81b94df14b
Rework readme, move code example to the top, link draft of Colab (#118) 2022-12-03 00:17:57 +04:00
Alexander Borzunov
7bd5916744
Make Petals a pip-installable package (attempt 2) (#102)
1. Petals can be now installed using `pip install git+https://github.com/bigscience-workshop/petals`
    - In case if you already cloned the repo, you can do `pip install .` or `pip install .[dev]`
2. Moved `src` => `src/petals`
    - Replaced `from src.smth import smth` with `from petals.smth import smth`
3. Moved `cli` => `src/petals/cli`
    - Replaced `python -m cli.run_smth` with `python -m petals.cli.run_smth` (all utilities are now available right after pip installation)
4. Moved the `requirements*.txt` contents to `setup.cfg` (`requirements.txt` for packages is not supported well by modern packaging utils)
5. Increased the package version from `0.2` to `1.0alpha1`
2022-11-30 10:41:13 +04:00
Alexander Borzunov
0c3781a89c
Shorten bullet points in readme 2022-11-30 07:20:27 +04:00
Alexander Borzunov
898f614515
Fix floating point issues in block_selection.py (#89) 2022-11-25 02:17:59 +04:00
Alexander Borzunov
c07a7e0812
Add "Terms of Use" 2022-11-21 18:54:07 +04:00
Artem Chumachenko
0d9c7de0bd
Add sst-2 ipynb example (#86)
- Add sst-2 example of a prompt-based training
- Have some enhancement in the persona-chat example
2022-11-07 13:55:00 +04:00
Alexander Borzunov
640bbc38a9
Make even smaller readability changes 2022-09-20 15:03:57 +04:00
Alexander Borzunov
d1b012b479
Make small readability & style changes to the instructions (#77) 2022-09-20 15:00:59 +04:00
justheuristic
fef48d7d99
Use bitsandbytes==0.34.0, update readme (#76)
* unlock bnb backward
* Fix bnb version in README
* Update requirements.txt
2022-09-20 13:07:34 +03:00
justheuristic
8caf1145a8
Quality of life changes: update readme, simplify run_server interface (#75)
- run_server now accepts model name as both positional and keyword argument
- changed names in README to account for interface updates
- moved model conversion from README to a separate wiki page
- updated requirements.txt
2022-09-20 03:51:57 +03:00
Alexander Borzunov
54ad745bed
Warn that current instructions involve 6B model but we will replace them soon (#63) 2022-09-05 15:05:59 +04:00
Alexander Borzunov
5f0c5329d4
Update readme with arxiv link and more discussions (#62)
Co-authored-by: justheuristic <justheuristic@gmail.com>
2022-09-05 12:04:50 +04:00
Alexander Borzunov
9bea7b9ea8
Update bullet points with feedback from Tim and other people (#61)
Co-authored-by: Tim Dettmers <tim.dettmers@gmail.com>
2022-09-03 06:38:18 +04:00
Alexander Borzunov
7653562aa1
Use latest version of Petals scheme, shrink Petals logo (#59) 2022-09-02 15:38:04 +04:00
Alexander Borzunov
2eb5843852
Update readme for the 1st public release (#57) 2022-09-01 08:41:49 +04:00
Pavel Samygin
0be21775af
remove transformer block, implement as sequential of size 1 (#54)
* remove transformer block, implement as sequence size 1
* reimplement get_remote_module
* fix readme

Co-authored-by: Alexander Borzunov <hxrussia@gmail.com>
Co-authored-by: Aleksandr Borzunov <borzunov.alexander@gmail.com>
2022-09-01 04:26:31 +03:00
justheuristic
d271b75dd4
Let users specify sequence length instead of assuming 2048 (#52)
- Maximum length is now provided in `.inference_session(max_length=100)`
   - previously, we would always assume max length = 2048
- added a generic way to forward **kwargs to inference session
  - for compatibility with #47 
  - Note to @borzunov : it does *not* pass them arbitrarily, but instead checks for kwarg names at the bottom level
- run_server can be started with a custom max_length for inference
- renamed --cache_size_bytes to --attention_cache_bytes (to avoid collision with --cache_dir)
- --attn_cache_bytes can now support humane file sizes (e.g. 300MB instead of 314572800)
- made some server-side errors more human-readable to user (e.g. when max length is exceeded)

Co-authored-by: Aleksandr Borzunov <borzunov.alexander@gmail.com>
Co-authored-by: Alexander Borzunov <hxrussia@gmail.com>
2022-08-29 21:04:37 +03:00
Dmitry Baranchuk
11a424837f
integrate mixed-8bit model (#39)
* integrate mixed-8bit model
* Fix bug with model duplication in RAM
* set throughput=1.0 to fix zero throughput problem
* add revision support
* update hivemind and bitsandbytes
* update deploy scripts
* update installation instructions
2022-08-04 09:57:37 +03:00
Alexander Borzunov
7d39d46966
Use "PETALS" as the readme title (#40)
Since we've chosen the system name, let's use it in the repo name and the readme title.
2022-08-02 18:48:54 +04:00
justheuristic
ccdcefe405
Add instructions to test the full model (#25)
add instructions to test the full model
2022-07-16 02:45:10 +03:00
justheuristic
eb0a6be716
Clean up readme (#24)
Remove some deprecated sections of README and turns on CI for main branch
2022-07-16 02:11:17 +03:00
Alexander Borzunov
7e9f337a63
Remove excess line from readme 2022-07-14 22:27:17 +04:00
Alexander Borzunov
aba43f1308
Implement block selection on servers (#20) 2022-07-12 14:42:30 +04:00
justheuristic
4eadd00a2c rm prefix from tests 2022-07-08 19:11:55 +03:00
justheuristic
2e90ac30a0 use default prefix in readme 2022-07-07 13:22:03 +03:00
justheuristic
4695071ad2 WIP: make DistributedBloom compliant with HF interface 2022-07-07 03:11:28 +03:00
justheuristic
8de7c1687b
list latest additions 2022-07-01 03:57:21 +03:00
justheuristic
d688cb0d22 stupid, slow, fragile, but correct full model inference 2022-07-01 03:53:08 +03:00
Aleksandr Borzunov
b78d713347 refactor, add swarm info 2022-06-29 14:26:47 +03:00
justheuristic
ca3c08acc1
Update README.md 2022-06-24 10:05:39 +03:00
justheuristic
1cdf8a77fb basic chained inference (multiple blocks per one RPC call) 2022-06-23 16:33:16 +03:00
justheuristic
0e7afea026 Merge remote-tracking branch 'origin/main' into main 2022-06-22 22:02:27 +03:00
justheuristic
2eb47cbedd support hosting multiple instances of the same block 2022-06-22 22:00:55 +03:00
justheuristic
14b6d04b0f install hivemind from pip 2022-06-22 17:32:13 +03:00