mirror of
https://github.com/huggingface/transformers.git
synced 2025-07-05 13:50:13 +06:00
e93103632b
* First commit * step 1 working * add alibi * placeholder for `scan` * add matrix mult alibi * beta scaling factor for bmm * working v1 - simple forward pass * move layer_number from attribute to arg in call * partial functioning scan * hacky working scan * add more modifs * add test * update scan for new kwarg order * fix position_ids problem * fix bug in attention layer * small fix - do the alibi broadcasting only once * prelim refactor * finish refactor * alibi shifting * incorporate dropout_add to attention module * make style * make padding work again * update * remove bogus file * up * get generation to work * clean code a bit * added small tests * adding albii test * make CI tests pass: - change init weight - add correct tuple for output attention - add scan test - make CI tests work * fix few nits * fix nit onnx * fix onnx nit * add missing dtype args to nn.Modules * remove debugging statements * fix scan generate * Update modeling_flax_bloom.py * Update test_modeling_flax_bloom.py * Update test_modeling_flax_bloom.py * Update test_modeling_flax_bloom.py * fix small test issue + make style * clean up * Update tests/models/bloom/test_modeling_flax_bloom.py Co-authored-by: Sanchit Gandhi <93869735+sanchit-gandhi@users.noreply.github.com> * fix function name * small fix test * forward contrib credits from PR17761 * Fix failing test * fix small typo documentation * fix non passing test - remove device from build alibi * refactor call - refactor `FlaxBloomBlockCollection` module * make style * upcast to fp32 * cleaner way to upcast * remove unused args * remove layer number * fix scan test * make style * fix i4 casting * fix slow test * Update src/transformers/models/bloom/modeling_flax_bloom.py Co-authored-by: Sanchit Gandhi <93869735+sanchit-gandhi@users.noreply.github.com> * remove `layer_past` * refactor a bit * fix `scan` slow test * remove useless import * major changes - remove unused code - refactor a bit - revert import `torch` * major refactoring - change build alibi * remove scan * fix tests * make style * clean-up alibi * add integration tests * up * fix batch norm conversion * style * style * update pt-fx cross tests * update copyright * Update src/transformers/modeling_flax_pytorch_utils.py Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com> * per-weight check * style * line formats --------- Co-authored-by: younesbelkada <younesbelkada@gmail.com> Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com> Co-authored-by: Younes Belkada <49240599+younesbelkada@users.noreply.github.com> Co-authored-by: haileyschoelkopf <haileyschoelkopf@users.noreply.github.com> Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
98 lines
3.7 KiB
Markdown
98 lines
3.7 KiB
Markdown
<!--Copyright 2022 The HuggingFace Team. All rights reserved.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
|
|
the License. You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
|
|
an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
|
|
specific language governing permissions and limitations under the License.
|
|
|
|
⚠️ Note that this file is in Markdown but contain specific syntax for our doc-builder (similar to MDX) that may not be
|
|
rendered properly in your Markdown viewer.
|
|
|
|
-->
|
|
|
|
# BLOOM
|
|
|
|
## Overview
|
|
|
|
The BLOOM model has been proposed with its various versions through the [BigScience Workshop](https://bigscience.huggingface.co/). BigScience is inspired by other open science initiatives where researchers have pooled their time and resources to collectively achieve a higher impact.
|
|
The architecture of BLOOM is essentially similar to GPT3 (auto-regressive model for next token prediction), but has been trained on 46 different languages and 13 programming languages.
|
|
Several smaller versions of the models have been trained on the same dataset. BLOOM is available in the following versions:
|
|
|
|
- [bloom-560m](https://huggingface.co/bigscience/bloom-560m)
|
|
- [bloom-1b1](https://huggingface.co/bigscience/bloom-1b1)
|
|
- [bloom-1b7](https://huggingface.co/bigscience/bloom-1b7)
|
|
- [bloom-3b](https://huggingface.co/bigscience/bloom-3b)
|
|
- [bloom-7b1](https://huggingface.co/bigscience/bloom-7b1)
|
|
- [bloom](https://huggingface.co/bigscience/bloom) (176B parameters)
|
|
|
|
## Resources
|
|
|
|
A list of official Hugging Face and community (indicated by 🌎) resources to help you get started with BLOOM. If you're interested in submitting a resource to be included here, please feel free to open a Pull Request and we'll review it! The resource should ideally demonstrate something new instead of duplicating an existing resource.
|
|
|
|
<PipelineTag pipeline="text-generation"/>
|
|
|
|
- [`BloomForCausalLM`] is supported by this [causal language modeling example script](https://github.com/huggingface/transformers/tree/main/examples/pytorch/language-modeling#gpt-2gpt-and-causal-language-modeling) and [notebook](https://colab.research.google.com/github/huggingface/notebooks/blob/main/examples/language_modeling.ipynb).
|
|
|
|
See also:
|
|
- [Causal language modeling task guide](../tasks/language_modeling)
|
|
- [Text classification task guide](../tasks/sequence_classification)
|
|
- [Token classification task guide](../tasks/token_classification)
|
|
- [Question answering task guide](../tasks/question_answering)
|
|
|
|
|
|
⚡️ Inference
|
|
- A blog on [Optimization story: Bloom inference](https://huggingface.co/blog/bloom-inference-optimization).
|
|
- A blog on [Incredibly Fast BLOOM Inference with DeepSpeed and Accelerate](https://huggingface.co/blog/bloom-inference-pytorch-scripts).
|
|
|
|
⚙️ Training
|
|
- A blog on [The Technology Behind BLOOM Training](https://huggingface.co/blog/bloom-megatron-deepspeed).
|
|
|
|
## BloomConfig
|
|
|
|
[[autodoc]] BloomConfig
|
|
- all
|
|
|
|
## BloomModel
|
|
|
|
[[autodoc]] BloomModel
|
|
- forward
|
|
|
|
## BloomTokenizerFast
|
|
|
|
[[autodoc]] BloomTokenizerFast
|
|
- all
|
|
|
|
## BloomForCausalLM
|
|
|
|
[[autodoc]] BloomForCausalLM
|
|
- forward
|
|
|
|
## BloomForSequenceClassification
|
|
|
|
[[autodoc]] BloomForSequenceClassification
|
|
- forward
|
|
|
|
## BloomForTokenClassification
|
|
|
|
[[autodoc]] BloomForTokenClassification
|
|
- forward
|
|
|
|
## BloomForQuestionAnswering
|
|
|
|
[[autodoc]] BloomForQuestionAnswering
|
|
- forward
|
|
|
|
## FlaxBloomModel
|
|
|
|
[[autodoc]] FlaxBloomModel
|
|
- __call__
|
|
|
|
## FlaxBloomForCausalLM
|
|
|
|
[[autodoc]] FlaxBloomForCausalLM
|
|
- __call__
|