Skip to content

Commit

Permalink
Squashed commit of the following:
Browse files Browse the repository at this point in the history 
commit 37c5ca5eb9012a1009cf23b892828902f6a8799a
Author: Raushan Turganbay <[email protected]>
Date:   Tue Aug 6 10:24:19 2024 +0500

    Cache: create docs (#32150)

    * draft

    * updates

    * works?

    * try adding python example in hidden section

    * another try

    * hwo do i render python

    * format as html code?

    * Update docs/source/en/kv_cache.md

    Co-authored-by: Joao Gante <[email protected]>

    * Update docs/source/en/kv_cache.md

    Co-authored-by: Joao Gante <[email protected]>

    * Update docs/source/en/kv_cache.md

    Co-authored-by: Joao Gante <[email protected]>

    * Update docs/source/en/kv_cache.md

    Co-authored-by: Joao Gante <[email protected]>

    * Update docs/source/en/kv_cache.md

    Co-authored-by: Joao Gante <[email protected]>

    * one more small update

    * should render hidden secrtion now

    * add outputs

    * fix links

    * check links

    * update all links

    * update with offloaded cache

    * all cache is importable, so they appear in docs

    * fix copies

    * docstring...

    ---------

    Co-authored-by: Joao Gante <[email protected]>

commit 13dc6b0853c3cb54e79b18105c0528bc9e84881c
Author: Francisco Kurucz <[email protected]>
Date:   Mon Aug 5 19:14:50 2024 -0300

    Fix documentation links and code reference to model llava-next (#32434)

commit 7e5d46ded433605a906fcab6be43ac85307cca9b
Author: amyeroberts <[email protected]>
Date:   Mon Aug 5 16:33:19 2024 +0100

    Respect the config's attn_implementation if set (#32383)

    * Respect the config's attn if set

    * Update test - can override in from_config

    * Fix

commit 458b0cd2c544cdd6c700f9b0c21077c889bcee6c
Author: Sai-Suraj-27 <[email protected]>
Date:   Mon Aug 5 19:49:42 2024 +0530

    fix: Updated `test_embeded_special_tokens` for luke and mluke models (#32413)

    Fixed tokenizertests for luke, mluke models.

commit baf7e5c927744122c89ab1270c6c312541c7eb41
Author: Abdi <[email protected]>
Date:   Mon Aug 5 21:15:36 2024 +0800

    Persist embedding type of BART and mBART models after resize (#32242)

    * fix: persist embedding type of MBartConditonalGeneration after resize

    * fix: persist embedding type of BartConditonalGeneration after resize

commit f5f1e52f6cf13cdf63ff25c311d33e2f2a842911
Author: Francisco Kurucz <[email protected]>
Date:   Mon Aug 5 05:18:28 2024 -0300

    Fix documentation references to google/bit-50 model (#32407)

commit ea5da52ebc062ff56f0e3aa05b0e3cc981731e14
Author: Nicholas Broad <[email protected]>
Date:   Mon Aug 5 00:51:58 2024 -0700

    add values for neftune (#32399)

    I always forget what typical values are, and I have to look at the paper everytime. This will be a helpful reminder.

commit 3d7c2f9dea45338b7ebcd459b452e2fad7abfa1f
Author: Ita Zaporozhets <[email protected]>
Date:   Mon Aug 5 09:22:48 2024 +0200

    * save total_vocab_size = vocab_size + user added tokens to speed up operation

    * updating length when added_tokens_decoder is set

    * add test len(tokenizer)

commit 3bb646a54f42030e9bafa47cd3f64367691a3bc5
Author: Raushan Turganbay <[email protected]>
Date:   Mon Aug 5 11:58:42 2024 +0500

    Phi3 tests: fix typing for Python 3.8 (#32388)

    fix phi

commit 05ae3a300d6f3534eeb99a08828a5bae6dd973db
Author: TechInterMezzo <[email protected]>
Date:   Mon Aug 5 08:40:58 2024 +0200

    fix: SeamlessM4TFeatureExtractor stride remainder (#32088)

    * fix: SeamlessM4TFeatureExtractor stride remainder

    * Added attention mask size test

    * Reran ruff for style correction

commit 847bb856d55e3664150e408448fa59d0705b4d60
Author: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
Date:   Mon Aug 5 08:38:34 2024 +0200

    Bump keras from 2.8.0 to 2.13.1 in /examples/research_projects/decision_transformer (#32393)

    Bump keras in /examples/research_projects/decision_transformer

    Bumps [keras](https://github.com/keras-team/keras) from 2.8.0 to 2.13.1.
    - [Release notes](https://github.com/keras-team/keras/releases)
    - [Commits](https://github.com/keras-team/keras/compare/v2.8.0...v2.13.1)

    ---
    updated-dependencies:
    - dependency-name: keras
      dependency-type: direct:production
    ...

    Signed-off-by: dependabot[bot] <[email protected]>
    Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>

commit 621fb3c0edddf98f3272f3b197e772af4fa30b6c
Author: Xueshen Liu <[email protected]>
Date:   Sat Aug 3 14:07:55 2024 -0400

    MixtralFlashAttention2: put "plus 1" inside parentheses when calculating rotary_seq_len, allowing None position_ids input. (#31500)

    * Mixtral: remove unnecessary plus 1 when calculating rotary_seq_len, allowing position_ids=None (no auto position_ids generation could be unsafe)

    * fix typo [:-1] to [:, -1]

    * to meet formatting requirement

    * to meet formatting requirement

    * remove white space

    * MixtralFlashAttention2: put "+ 1" inside parentheses when calculating rotary_seq_len, allowing None position_ids input. Fix format/style issue.

    * propagate to startcoder2, phi3, mixtral and qwen2

    * update qwen2_moe

commit 7c31d05b59a9dce24b8ddc4b2bb8c8cf6bb5fd77
Author: Shaopeng Fu <[email protected]>
Date:   Sat Aug 3 19:24:11 2024 +0300

    fix: (issue #32124) Exception raised when running `transformers/examples/flax/language-modeling/t5_tokenizer_model.py`. (#32157)

    fix: Exception raised when running .

commit c1aa0edb48217f416f4bbe6e3a9db1500284513b
Author: Sanchit Gandhi <[email protected]>
Date:   Fri Aug 2 17:32:50 2024 +0800

    [generate] only require an attention mask for mps with torch<2.4 (#32367)

    * up

    * style

    * stopping

commit 083e13b7c47f674b11c74d1b7c7ee7cd1241b406
Author: Joao Gante <[email protected]>
Date:   Fri Aug 2 09:39:45 2024 +0100

    RoPE: Add numerical tests ✨  (#32380)

    tests! :D

commit 2af199c42b545f6248475ce456dd6c2a351b8522
Author: Raushan Turganbay <[email protected]>
Date:   Fri Aug 2 09:54:16 2024 +0500

    Update docs (#32368)

    nits

commit 82efc53513a51660e629c7eca8210af1d67df00b
Author: Zach Mueller <[email protected]>
Date:   Thu Aug 1 15:18:43 2024 -0400

    Yell at the user if zero-3 init wasn't performed, but expected to have been done (#32299)

    * Test this zach

    * Test for improper init w/o zero3

    * Move back

    * Apply suggestions from code review

    Co-authored-by: amyeroberts <[email protected]>

    * Get rid of stars in warning

    * Make private

    * Make clear

    ---------

    Co-authored-by: amyeroberts <[email protected]>

commit 51ab25e2932da15511ced35bcbdfa92d25c4794c
Author: OsamaS99 <[email protected]>
Date:   Thu Aug 1 14:57:42 2024 +0200

    Fixed Hybrid Cache Shape Initialization. (#32163)

    * fixed hybrid cache init, added test

    * Fix Test Typo

    ---------

    Co-authored-by: Aaron Haag <[email protected]>

commit e3d8285a84f803e962050e2c2283f3362e36bfbc
Author: Joao Gante <[email protected]>
Date:   Thu Aug 1 13:46:11 2024 +0100

    Docker: add `speech` dep to the consistency docker image (#32374)

commit ca59d6f77c9fda197222f9aa9205d8c7b5dff34e
Author: Nikos Karampatziakis <[email protected]>
Date:   Thu Aug 1 05:42:07 2024 -0700

    Offloaded KV Cache (#31325)

    * Initial implementation of OffloadedCache

    * enable usage via cache_implementation

    * Address feedback, add tests, remove legacy methods.

    * Remove flash-attn, discover synchronization bugs, fix bugs

    * Prevent usage in CPU only mode

    * Add a section about offloaded KV cache to the docs

    * Fix typos in docs

    * Clarifications and better explanation of streams

commit b4727a1216bb21df2795e973063ed07202235d7e
Author: Omar Salman <[email protected]>
Date:   Thu Aug 1 17:32:13 2024 +0500

    Fix conflicting key in init kwargs in PreTrainedTokenizerBase (#31233)

    * Fix conflicting key in init kwargs in PreTrainedTokenizerBase

    * Update code to check for callable key in save_pretrained

    * Apply PR suggestions

    * Invoke CI

    * Updates based on PR suggestion

commit db8c7caeb6b3969a2153b36ba3e5fdef6534c1d6
Author: Viktor Scherbakov <[email protected]>
Date:   Thu Aug 1 14:30:10 2024 +0200

    Empty list in defaults for LLaMA special tokens during weights conversion (#32342)

    empty list in defaults

commit 2229ebe7220fb54bc5f91f575c2d7a988e7122cb
Author: Ita Zaporozhets <[email protected]>
Date:   Thu Aug 1 13:57:41 2024 +0200

    update clean_up_tokenization_spaces warning (#32371)

commit 05c1f9af9a5ebd213dd923e97f6fbed4c115f3c6
Author: Hanna Yukhymenko <[email protected]>
Date:   Thu Aug 1 13:52:05 2024 +0200

    Check device map for saving tokenizer config on TPU (fix for issue #31971) (#32043)

    * Remove TPU device map for saving tokenizer config

    * Update tokenization_utils_base.py

    * Fix error msg when passing non-string device into tokenizer

    * Fix error message for non-string tokenizer device

    * Print out tokenizer device type in error msg

    * Update tokenization_utils_base.py

commit 9e2828403218da16d9759c9be020b70f51df373d
Author: nv-guomingz <[email protected]>
Date:   Thu Aug 1 19:51:20 2024 +0800

    add missing attribute _supports_param_buffer_assignment for gpt-j. (#32359)

    Co-authored-by: Guoming Zhang <[email protected]>

commit 48ed24c50ab29bf690f2ab030721e6a8b0aa5205
Author: Lunwen He <[email protected]>
Date:   Thu Aug 1 04:49:00 2024 -0700

    Remove size check between attn_weights and kv_seq_len for phi3 (#32339)

    * Remove size check between attn_weights and kv_seq_len

    * add unit tests

commit e234061cddd28bb8b82144833241883816289e40
Author: Sanchit Gandhi <[email protected]>
Date:   Thu Aug 1 18:10:56 2024 +0800

    [whisper] compile compatibility with long-form decoding (#31772)

    * [whisper] compile compatibility with long-form decoding

    * clarify comment

    * fix after rebase

    * finalise

    * fix bsz

    * fix cache split

    * remove contiguous

    * style

    * finish

    * update doc

    * prevent cuda graph trace

commit 9451a385261b30e7319a2c93285ab76161e8c003
Author: Sanchit Gandhi <[email protected]>
Date:   Thu Aug 1 16:05:27 2024 +0800

    [enc-dec cache] fix bug in indexing (#32370)

commit 453e74884fb7e2613e7b45033fbb3c1cadb638b4
Author: Raushan Turganbay <[email protected]>
Date:   Thu Aug 1 09:48:03 2024 +0500

    LLaVa: add cache class attribute (#32278)

    cache class flag

commit 14ee2326e51cb210cec72f31b248cb722e9d5d1f
Author: Ricardo <[email protected]>
Date:   Thu Aug 1 06:34:22 2024 +0800

    fix: warmup_steps check for training_args (#32236)

commit 53f0c9c2906e0b0f1623bfdfb420fca1e655098d
Author: Sai-Suraj-27 <[email protected]>
Date:   Thu Aug 1 01:26:50 2024 +0530

    fix: Removed unnecessary `@staticmethod` decorator (#32361)

    * Fixed staticmethods with self as first argument.

    * Fixed staticmethods with self as first argument.

    * Fixed staticmethods with self as first argument.

    * Fixed staticmethods with self as first argument.

commit 92abe6033491dcaa958235e551f40f6b417d3771
Author: fxmarty <[email protected]>
Date:   Wed Jul 31 20:03:07 2024 +0200

    >3-5x faster torch.compile forward compilation for autoregressive decoder models (#32227)

    * draft

    * apply changes to all relevant archs

    * rerun ci - check_docstrings.py failing?

    * fix docstring

    * move 2D->4D mask creation to modeling file

    * repo consistency

    * fix the batch size = 1 case - calling contiguous is not enough

    * nit

    * style

    * propagate to gemma/gemma-2

    * prepare inputs for gemma generation

    * implement test and tiny fix in gemma2

    * Update src/transformers/models/bloom/modeling_bloom.py

    Co-authored-by: Arthur <[email protected]>

    * fix copies

    * ci pass

    * fix gemma's test_compile_static_cache tests

    * flacky

    * retrigger ci

    ---------

    Co-authored-by: sanchit-gandhi <[email protected]>
    Co-authored-by: Arthur <[email protected]>

commit b46bd8b9d2ac991c0c04674957ebc0a65fb3f42b
Author: Aymeric Roucher <[email protected]>
Date:   Wed Jul 31 18:44:53 2024 +0200

    Fix error when streaming to gradio with non-string tool arguments (#32360)

    Fix error when streaming agent run to gradio with non-string tool arguments

commit ef177a5e1cdf0ca53e24e6d76e813198f7300dc4
Author: Joao Gante <[email protected]>
Date:   Wed Jul 31 16:04:48 2024 +0100

    Gemma 2: support assisted generation (#32357)

commit 5f1fcc299cb00c1edce5eb1efb8bacdde2365690
Author: amyeroberts <[email protected]>
Date:   Wed Jul 31 14:51:04 2024 +0100

    [Idefics2] - Fix FA2 call for Perceiver layer (#32275)

    * Fix FA2 call for Perciever layer

    * [run_slow] idefics2

    * [run_slow] idefics2

    * [run_slow] idefics2

    * Fix up

    * [run_slow] idefics2

    * [run_slow] idefics2

    * [run_slow] idefics2

commit b75ad56620431984a44a962c98136c8571b4fca9
Author: Joao Gante <[email protected]>
Date:   Wed Jul 31 11:12:46 2024 +0100

    Llama 3.1: Fix incorrect `inv_freq` assignment (#32330)

    fix 💩

commit 7f552e28e0aca00ce60868c7620f7463eab60e14
Author: Raushan Turganbay <[email protected]>
Date:   Wed Jul 31 10:33:38 2024 +0500

    Gemma2 and flash-attention (#32188)

    * enable flash-attn & static cache

    * this works, not the prev

    * fix for sliding window layers

    * not needed anymore

commit a3264332cfb5ab8675ddb42740a75aeee1782a74
Author: Raushan Turganbay <[email protected]>
Date:   Wed Jul 31 10:01:12 2024 +0500

    LLaVA-NeXT: fix anyres shapes (#32314)

    fix

commit 6e2d04e429dc4ce240c99bd14b7b84550b79fd73
Author: Joshua Lochner <[email protected]>
Date:   Tue Jul 30 23:36:38 2024 +0200

    Fix slow GemmaTokenizer and improve SPM slow -> fast conversion process (#32191)

    * Remove user-defined tokens which can be obtained through merges

    * Remove debug line

    * formatting

    * Refactor spm slow -> fast converter

    * revert unnecessary refactor

    * set comprehension

    * remove test files

    * Use `vocab_scores`

    * Always replace spiece underline with space in decode

    * we no longer need token filtering

    * Add save fast load slow unit test

    * Remove tokenizers version check

    * Remove duplicate code

    * Make `<start_of_turn>` and `<end_of_turn>` special tokens

    * Bias merge priority with length if score is the same

    * Add unit test for merge priority

    * CI

commit 026a173a64372e9602a16523b8fae9de4b0ff428
Author: Joao Gante <[email protected]>
Date:   Tue Jul 30 18:56:10 2024 +0100

    Repo checks: skip docstring checks if not in the diff (#32328)

    * tmp

    * skip files not in the diff

    * use git.Repo instead of an external subprocess

    * add tiny change to confirm that the diff is working on pushed changes

    * add make quality task

    * more profesh main commit reference

commit 516af4bb63538edc448f814e3690dd5171c4f311
Author: fkrasnov2 <[email protected]>
Date:   Tue Jul 30 20:21:45 2024 +0300

    fixes #32329 : The Torch code is correct - to get an average of 10% o… (#32335)

    fixes #32329 : The Torch code is correct - to get an average of 10% of the total, we want to take 50% of the remainder after we've already masked 80% with [MASK] in the previous step.

commit 62c60a30181a65e1a3a7f19c3055a240a6a21335
Author: Wing Lian <[email protected]>
Date:   Tue Jul 30 12:55:59 2024 -0400

    fixes to properly shard FSDP across cpu and meta for cpu_efficient_loading for prequantized 4bit (#32276)

commit 16271080333ad52be5349fb31d789fb232b68760
Author: Sai-Suraj-27 <[email protected]>
Date:   Tue Jul 30 22:23:03 2024 +0530

    fix: Added missing raise keyword for few exceptions (#32333)

    Fixed raising of few exceptions.

commit bd54ed2ed7f578e4122f3e6d536fbe3c9bc76de1
Author: plaggy <[email protected]>
Date:   Tue Jul 30 18:48:18 2024 +0200

    Alternative agent plan (#32295)

    * new agent plan

    * plan type assertion

    * style corrections

    * better prompt naming

    * make fixup

commit e68ec18ce224af879f22d904c7505a765fb77de3
Author: Joao Gante <[email protected]>
Date:   Tue Jul 30 15:49:14 2024 +0100

    Docs: formatting nits (#32247)

    * doc formatting nits

    * ignore non-autodocs

    * Apply suggestions from code review

    Co-authored-by: amyeroberts <[email protected]>

    * Update src/transformers/models/esm/modeling_esm.py

    Co-authored-by: amyeroberts <[email protected]>

    * Update src/transformers/models/esm/modeling_esm.py

    Co-authored-by: amyeroberts <[email protected]>

    * make fixup

    ---------

    Co-authored-by: amyeroberts <[email protected]>

commit 2fbbcf5007509c66b02924ce6dcff66f58e7f58c
Author: Yoach Lacombe <[email protected]>
Date:   Tue Jul 30 16:00:13 2024 +0200

    Fix M4T for ASR pipeline (#32296)

    * tentative fix

    * do the same for M4T

commit 084b5094eb490319719cc11cb05b751e0b419d49
Author: Luc Georges <[email protected]>
Date:   Tue Jul 30 14:49:26 2024 +0200

    feat(ci): set `fetch-depth: 0` in trufflehog checkout step (#31663)

commit 20528f067cf9204cea5178ce0f837245e146e159
Author: Teddy Ferdinan <[email protected]>
Date:   Tue Jul 30 11:25:54 2024 +0200

    Cast epochs_trained to int when resuming training (#32286)

    * fix epochs_trained as int when resuming training

    * refactor

    ---------

    Co-authored-by: teddyferdinan <[email protected]>

commit 934fe1504e6d5e87e01d96305f4d97faa63cf4c1
Author: Isotr0py <[email protected]>
Date:   Tue Jul 30 17:01:00 2024 +0800

    Fix GGUF dequantize for `gguf==0.9.1` (#32298)

    * fix gguf dequantize for gguf==0.9.1

    * fix old version

    * make style

commit 3e8106d2533cbd890ddd1e919bd62132cd4718c3
Author: Gilad Turok <[email protected]>
Date:   Tue Jul 30 03:19:24 2024 -0400

    Docs: fix GaLore optimizer code example (#32249)

    Docs: fix GaLore optimizer example

    Fix incorrect usage of GaLore optimizer in Transformers trainer code example.

    The GaLore optimizer uses low-rank gradient updates to reduce memory usage. GaLore is quite popular and is implemented by the authors in [https://github.com/jiaweizzhao/GaLore](https://github.com/jiaweizzhao/GaLore). A few months ago GaLore was added to the HuggingFace Transformers library in https://github.com/huggingface/transformers/pull/29588.

    Documentation of the Trainer module includes a few code examples of how to use GaLore. However, the `optim_targe_modules` argument to the `TrainingArguments` function is incorrect, as discussed in https://github.com/huggingface/transformers/pull/29588#issuecomment-2006289512. This pull request fixes this issue.

commit f0bc49e7f61f74f055c47ad40e6010f57eed0b0b
Author: Yih-Dar <[email protected]>
Date:   Mon Jul 29 22:12:21 2024 +0200

    use torch 2.4 in 2 CI jobs (#32302)

    Co-authored-by: ydshieh <[email protected]>

commit a24a9a66f446dcb9277e31d16255536c5ce27aa6
Author: Aymeric Roucher <[email protected]>
Date:   Mon Jul 29 20:12:44 2024 +0200

    Add stream messages from agent run for gradio chatbot (#32142)

    * Add stream_to_gradio method for running agent in gradio demo

commit 811a9caa2141bc98f96b36c69abcf1f934bd1fd2
Author: Guang Yang <[email protected]>
Date:   Mon Jul 29 10:19:15 2024 -0700

    Make static cache compatible with torch.export (#32168)

commit 7f5d644e69068825bb5b6e84cdc56b3d3a9bd04f
Author: Sanchit Gandhi <[email protected]>
Date:   Mon Jul 29 21:24:42 2024 +0800

    [pipeline] fix padding for 1-d tensors (#31776)

    * [pipeline] fix padding for 1-d tensors

    * add test

    * make style

    * Update tests/pipelines/test_pipelines_automatic_speech_recognition.py

    Co-authored-by: Kamil Akesbi <[email protected]>

    * Update tests/pipelines/test_pipelines_automatic_speech_recognition.py

    ---------

    Co-authored-by: Kamil Akesbi <[email protected]>

commit 3fbaaaa64d1ef3d8327adb577994d3d11277c77a
Author: Kamil Akesbi <[email protected]>
Date:   Mon Jul 29 11:19:52 2024 +0100

    Whisper tokenizer word level timestamps (#32197)

    * fix _fix_key in PreTrainedModel

    * fix _find_longest_common_sequence

    * add test

    * remove result.json

    * nit

    * update test

commit 7ffe25f2b935dcaf65079b04c5f91c8a42a99e28
Author: Joao Gante <[email protected]>
Date:   Mon Jul 29 10:52:13 2024 +0100

    Generate: end-to-end compilation (#30788)

    * mvp

    * added test (a few models need fixes)

    * fix a few test cases

    * test nits

    * harder test 😈

    * revert changes in stablelm

    * test with improved condition

    * add todo

    * tmp commit

    * merged with main

    * nits

    * add todo

    * final corrections

    * add docs for generation compilation

    * docs nits

    * add  tip

    * PR suggestions

    * add more details to the compilation docs

    * fix cache positions

    * cache is now init in generate; update docs

    * tag test as flaky

    * docs

    * post rebase make fixup and other nits

    * remove unintended changes

    * whisper (encoder-decoder) not supported

    * move token default updates to ; add tests for token defaults

    * push changes

    * manual rebase

    * chameleon doesn't support this

    * fix test_static_cache_mha_mqa_gqa (broken in another PR)

    * docs: dynamic is better with end-to-end compilation

commit 49928892d6491ff5a49c12cbc34695f6fa7ac0ed
Author: Sai-Suraj-27 <[email protected]>
Date:   Mon Jul 29 15:20:43 2024 +0530

    fix(docs): Fixed a link in docs (#32274)

    Fixed a link in docs.

commit 6494479f1de9fe16e9c6f89e52eb0cf81f864a7c
Author: Fanli Lin <[email protected]>
Date:   Mon Jul 29 17:29:11 2024 +0800

    make `p_mask` a numpy array before passing to `select_starts_ends` (#32076)

    * fix

    * bug fix

    * refine

    * fix

commit 535fe78b9f1d148684723e51f00645351880c47a
Author: Joao Gante <[email protected]>
Date:   Mon Jul 29 10:06:05 2024 +0100

    Repo: remove exceptions in `check_docstrings` (#32259)

    remove exceptions

commit a2ad9d5ad53f68c1ad268f7f46538eac6f5b631b
Author: Sai-Suraj-27 <[email protected]>
Date:   Mon Jul 29 14:13:09 2024 +0530

    fix: Fixed wrong argument passed to `convert_blip_checkpoint` function call (#32262)

    Removed one wrong argument passed to convert_blip_checkpoint function call.

commit 5019aabfacf7599b9a6b4e7a1adc1fb5c9017727
Author: leejet <[email protected]>
Date:   Mon Jul 29 15:51:43 2024 +0800

    Optimize t5 tokenize logic to avoid redundant calls (#32270)

    * Optimize t5 tokenize logic to avoid redundant calls

    * fix and overwrite copies

commit f2122cc6eb8e50e4d1b45da54b43bba59a458b30
Author: Yih-Dar <[email protected]>
Date:   Mon Jul 29 09:42:54 2024 +0200

    Upload new model failure report to Hub (#32264)

    upload

    Co-authored-by: ydshieh <[email protected]>

commit f7396876849926afa87c9412d67c43618dad403d
Author: Raushan Turganbay <[email protected]>
Date:   Mon Jul 29 10:58:59 2024 +0500

    🚨 Bloom support for cache class (#31445)

    * bloom dynamic cache

    * bloom follows standard cache format

    * no skips for bloom anymore

    * use cache position when possible

    * clean up

    * codestyle

    * Update src/transformers/models/bloom/modeling_bloom.py

    Co-authored-by: amyeroberts <[email protected]>

    * Update src/transformers/models/bloom/modeling_bloom.py

    Co-authored-by: amyeroberts <[email protected]>

    * Update src/transformers/models/bloom/modeling_bloom.py

    Co-authored-by: amyeroberts <[email protected]>

    * pr comments

    * isinstance fix

    * address comments

    * make musicgen test happy

    * [run-slow] bloom

    ---------

    Co-authored-by: amyeroberts <[email protected]>

commit 44f6fdd74f84744b159fa919474fd3108311a906
Author: Joao Gante <[email protected]>
Date:   Sat Jul 27 10:19:46 2024 +0100

    Llama 3.1: replace for loop by tensor ops at inv_freq initialization (#32244)

    * replace for loop by tensor ops

    * rm assert; readability

commit 8da90687308a10b33c5553b8a506cc04aab31702
Author: Yih-Dar <[email protected]>
Date:   Fri Jul 26 20:52:45 2024 +0200

    More flexible trigger condition (#32251)

    update

    Co-authored-by: ydshieh <[email protected]>

commit 81233c069c166af033794134bd8888783ac49ebe
Author: Raushan Turganbay <[email protected]>
Date:   Fri Jul 26 14:45:55 2024 +0500

    Flash-Attn: fix generation when no attention mask or no pading (#32241)

    * fix

    * fix prev test (half of failures)

    * [run-slow] llama, gemma2

    * [run-slow] llama, gemma2

commit 27c7f971c0dcd3bb423ea221fe2bce751d313119
Author: Fanli Lin <[email protected]>
Date:   Fri Jul 26 17:41:27 2024 +0800

    [tests] fix `static` cache implementation is not compatible with `attn_implementation==flash_attention_2` (#32039)

    * add flash attention check

    * fix

    * fix

commit 5f841c74b62754f186a8c06a684d491524b7bc03
Author: Connor Anderson <[email protected]>
Date:   Fri Jul 26 05:05:46 2024 -0400

    Add check for `target_sizes is None` in `post_process_image_guided_detection` for owlv2 (#31934)

    * Add check for target_sizes is None in post_process_image_guided_detection

    * Make sure Owlvit and Owlv2 in sync

    * Fix incorrect indentation; add check for correct size of target_sizes

commit f9756d9edb23354e3df50f7eb3f6b3129a25e453
Author: Rohit Dwivedula <[email protected]>
Date:   Fri Jul 26 04:05:38 2024 -0500

    Adds: extra_repr for RMSNorm layers in most models (#32204)

    * adds: extra_repr() to RMSNorm layers in multiple models

    * adds: extra_repr for deprecated models as well

    * formatting as per style guide

commit b8e5cd5396f7c0cc2d5e10be6696ea38742abf51
Author: Sai-Suraj-27 <[email protected]>
Date:   Fri Jul 26 14:03:02 2024 +0530

    Refactor: Removed un-necessary `object` base class (#32230)

    * Refactored to remove un-necessary object base class.

    * small fix.

commit 1c7ebf1d6eaf0ed0fd4101fd6eb7e64601429cfe
Author: João Nadkarni <[email protected]>
Date:   Fri Jul 26 09:38:59 2024 +0200

    don't log base model architecture in wandb if log model is false (#32143)

    * don't log base model architecture in wandb is log model is false

    * Update src/transformers/integrations/integration_utils.py

    Co-authored-by: amyeroberts <[email protected]>

    * convert log model setting into an enum

    * fix formatting

    ---------

    Co-authored-by: amyeroberts <[email protected]>

commit c46edfb8230bcc3152e8338742dc4822289acb3d
Author: Raushan Turganbay <[email protected]>
Date:   Fri Jul 26 10:52:06 2024 +0500

    Resize embeds with DeepSpeed  (#32214)

    * fix resize when deepspeed

    * deepsped uses new embeds

    * we needed this

commit fad15fba78e4603cd20695757ad899a6687485f9
Author: Raushan Turganbay <[email protected]>
Date:   Fri Jul 26 10:17:27 2024 +0500

    Llava: generate without images (#32183)

    * llava w/o images

    * tests

commit 4ab33c2d81866d4dd2f29df07f1a35491acbb39b
Author: Raushan Turganbay <[email protected]>
Date:   Fri Jul 26 10:16:06 2024 +0500

    Generation: stop at `eos` for assisted decoding (#31301)

    * fix

    * move changes to prompt lookup

    * add test

    * set eos in assistant model

    * style

    * fix flakiness

    * changes for new `main`

    * Update tests/generation/test_utils.py

    Co-authored-by: amyeroberts <[email protected]>

    * Update tests/generation/test_utils.py

    Co-authored-by: amyeroberts <[email protected]>

    * add comment to explain

    ---------

    Co-authored-by: amyeroberts <[email protected]>

commit 9d6c0641c4a3c2c5ecf4d49d7609edd5b745d9bc
Author: Pavel Iakubovskii <[email protected]>
Date:   Thu Jul 25 19:20:47 2024 +0100

    Fix code snippet for Grounding DINO (#32229)

    Fix code snippet for grounding-dino

commit 3a83ec48a63a8298c8193be48cf00785674bfb70
Author: jrhe <[email protected]>
Date:   Thu Jul 25 17:16:13 2024 +0100

    Allow a specific microphone to be used by the ffmpeg audio pipeline utility functions. Default to using the currently active microphone on Mac (#31846)

    * use currently active microphone on mac for ffmpeg_microphone

    * Allow ffmpeg_microphone device to be specified

    Co-authored-by: amyeroberts <[email protected]>

    ---------

    Co-authored-by: amyeroberts <[email protected]>

commit 6ed0bf1e8543a7d8e6640bbf9a655c5e1401f7de
Author: Huazhong Ji <[email protected]>
Date:   Fri Jul 26 00:01:06 2024 +0800

    translate philosophy.md to chinese (#32177)

    * translate philosophy.md to chinese

    * add the missing link

commit df6eee9201e4ba2b80cea021a18e95ada26ca2cc
Author: Yih-Dar <[email protected]>
Date:   Thu Jul 25 16:12:23 2024 +0200

    Follow up for #31973 (#32025)

    * fix

    * [test_all] trigger full CI

    ---------

    Co-authored-by: ydshieh <[email protected]>

commit de2318894e4f971ea2273c653a702dc93db2bd6a
Author: Kashif Rasul <[email protected]>
Date:   Thu Jul 25 15:12:23 2024 +0200

    [warnings] fix E721 warnings (#32223)

    fix E721 warnings

commit 9b9a54e61bf8749588178b37c23d77b90679fd10
Author: Kashif Rasul <[email protected]>
Date:   Thu Jul 25 15:11:43 2024 +0200

    [BigBird Pegasus] set _supports_param_buffer_assignment to False (#32222)

    set _supports_param_buffer_assignment to False

commit 1ecedf1d9ee927bac5b5bae8cb1892d936a5b622
Author: Austin <[email protected]>
Date:   Thu Jul 25 07:20:27 2024 -0500

    Update question_answering.py (#32208)

commit f53a5dec7b03eb195dc89c82ae761b033db1ceb6
Author: Huazhong Ji <[email protected]>
Date:   Thu Jul 25 17:04:04 2024 +0800

    remove unnecessary guard code related with pytorch versions 1.4.2 ~ 1.7.0 (#32210)

    remove unnecessary guard code related with pytorch versions 1.4.2 ~
    1.7.0

commit 5658e749adbaaf883caec003cecae8ce0a4261a6
Author: Sanchit Gandhi <[email protected]>
Date:   Thu Jul 25 16:58:02 2024 +0800

    [whisper] fix short-form output type (#32178)

    * [whisper] fix short-form output type

    * add test

    * make style

    * update long-form tests

    * fixes

    * last fix

    * finalise test

commit 85a1269e19af022e04bc2aad82572cd5a9e8cdd9
Author: Sai-Suraj-27 <[email protected]>
Date:   Wed Jul 24 22:30:21 2024 +0530

    fix: Replaced deprecated `unittest method` with the correct one (#32198)

    Replaced deprecated unittest method with the correct one.

commit edd68f4ed8db241bd3e9dc6c4ed96d471f243c9a
Author: Matt <[email protected]>
Date:   Wed Jul 24 17:36:32 2024 +0100

    :rotating_light: No more default chat templates (#31733)

    * No more default chat templates

    * Add the template to the GPT-SW3 tests since it's not available by default now

    * Fix GPT2 test

    * Fix Bloom test

    * Fix Bloom test

    * Remove default templates again

commit 1c122a46dc3c4448901f8d2f3018d9d58b846ba5
Author: Penut Chen <[email protected]>
Date:   Wed Jul 24 23:59:59 2024 +0800

    Support dequantizing GGUF FP16 format (#31783)

    * support gguf fp16

    * support gguf bf16 with pytorch

    * add gguf f16 test

    * remove bf16

commit af0e4b7b37b2d7eefe7531cf5201a5d6bae85525
Author: Marc Sun <[email protected]>
Date:   Wed Jul 24 17:14:05 2024 +0200

    Fix float8_e4m3fn in modeling_utils (#32193)

    * Fix float8_e4m3fn in modeling_utils

    * style

    * fix

    * comment

commit 1392a6867f40a55dfabaf306745c67627598b1af
Author: Raushan Turganbay <[email protected]>
Date:   Wed Jul 24 19:26:20 2024 +0500

    Fix resize embedding with Deepspeed (#32192)

    fix resize when deepspeed

commit 8d2534c4d0ab94a97a72d2ce6bb9ccd201abadb3
Author: Arthur <[email protected]>
Date:   Wed Jul 24 16:06:39 2024 +0200

    let's not warn when someone is running a forward  (#32176)

    * let's not warn when someone is running a foward without cache + self.training

    * more models

    * fixup

commit e0182f3bd7f4753c1e378e052ceea67898d97359
Author: Joao Gante <[email protected]>
Date:   Wed Jul 24 15:00:48 2024 +0100

    RoPE: relaxed rope validation (#32182)

    * relaxed rope check

    * lets also accept rope_type=None, defaulting to the original implementation

    * type and rope_type can coexist

commit 165116bc145dcc186fa287e624b28a9ab3a79955
Author: amyeroberts <[email protected]>
Date:   Wed Jul 24 14:03:40 2024 +0100

    Remove conversational pipeline tests (#32099)

    Remove conversation pipeline tests

commit 5f4ee98a7ade33e1c54fdd6181d04ee7b426b392
Author: Dr. Artificial曾小健 <[email protected]>
Date:   Wed Jul 24 18:54:41 2024 +0800

    Update qwen2.md (#32108)

    * Update qwen2.md

    outdated description

    * Update qwen2.md

    amended

    * Update qwen2.md

    Update

    * Update qwen2.md

    fix wrong version code, now good to go

commit 8678879f1dc2578cec18232146bf19de97aecaa1
Author: 조준래 <[email protected]>
Date:   Wed Jul 24 19:38:49 2024 +0900

    fix: default value reflects the runtime environment variables rather than the ones present at import time. (#32153)

    * fix: default value reflects the runtime environment variables rather than the ones present at import time.

    * Fix: Change `deterministic` to None by default; use env var if None

commit 01be5b48790f113b7d71943b580c842e3e097988
Author: Rohit Dwivedula <[email protected]>
Date:   Wed Jul 24 02:09:59 2024 -0500

    adds: extra_repr() to MambaRMSNorm to include hidden size / size of weights in the layer (#32171)

    * adds: extra_repr() to MambaRMSNorm to include the hidden size of the layer

    * style fix with ruff:

commit c85510f958e6955d88ea1bafb4f320074bfbd0c1
Author: Fanli Lin <[email protected]>
Date:   Wed Jul 24 00:47:51 2024 +0800

    [docs] change temperature to a positive value (#32077)

    fix

commit bc2adb0112b6677b0dfb4105c74570a0f92183eb
Author: Sai-Suraj-27 <[email protected]>
Date:   Tue Jul 23 21:22:41 2024 +0530

    fix: Fixed an if condition that is always evaluating to true (#32160)

    Fixed an if condition always evaluating to true.

commit 23f6a43f82fb2980f4b30cf3f95eb3a940384895
Author: Joao Gante <[email protected]>
Date:   Tue Jul 23 16:48:16 2024 +0100

    fix (#32162)

commit d5a99dfcee6e94065cb7c83cc8ab6fc5daa0cc4e
Author: Lysandre <[email protected]>
Date:   Tue Jul 23 16:58:17 2024 +0200

    Llama 3.1 conversion

    Co-authored-by: Arthur Zucker <[email protected]>

commit ff0d708fe627d6715f9a3e97d0a7947f70437447
Author: Lysandre <[email protected]>
Date:   Tue Jul 23 17:12:47 2024 +0200

    Dev version: v4.44.0.dev0

commit d2c687b3f1859b5c61258af14abba5312c0e6201
Author: Sai-Suraj-27 <[email protected]>
Date:   Tue Jul 23 20:37:31 2024 +0530

    Updated `ruff` to the latest version (#31926)

    * Updated ruff version and fixed the required code accorindg to the latest version.

    * Updated ruff version and fixed the required code accorindg to the latest version.

    * Added noqa directive to ignore 1 error shown by ruff

commit 9cf4f2aa9a9cecbb22e813931ef3bb72fc773540
Author: RhuiDih <[email protected]>
Date:   Tue Jul 23 21:56:41 2024 +0800

    Enhancing SFT Training Efficiency Using Packing and FlashAttention2 with Position IDs (#31629)

    * add DataCollatorBatchFlattening

    * Update data_collator.py

    * change name

    * new FA2 flow if position_ids is provided

    * add comments

    * minor fix

    * minor fix data collator

    * add test cases for models

    * add test case for data collator

    * remove extra code

    * formating for ruff check and check_repo.py

    * ruff format

    ruff format tests src utils

    * custom_init_isort.py

commit 7d92009af647167bae338e9d4af8bc0452c62fbf
Author: Deep Gandhi <[email protected]>
Date:   Tue Jul 23 19:11:52 2024 +0530

    Added additional kwarg for successful running of optuna hyperparameter search (#31924)

    Update integration_utils.py

    Added additional kwarg

commit 63700628adb91600c84fe3bbbc4c667cd3e3aa71
Author: Alvaro Moran <[email protected]>
Date:   Tue Jul 23 14:18:19 2024 +0200

    feat(cache): StaticCache uses index_copy_ to avoid useless copy (#31857)

    * feat(cache): StaticCache uses index_copy_ to avoid useless copy

    Using index_copy_ allows for explicit in-place change of the tensor.
    Some backends (XLA) will otherwise copy the tensor, making the code
    slower and using more memory.

    Proposed implementation will end up using less memory and on XLA will
    result in less compilation, but the change is also quite generic, making
    no change whatsoever on CUDA or CPU backend.

    * feat(cache): SlidingWindowCache uses index_copy_ to avoid useless copy

    Applying the same change done in StaticCache.

    * fix(cache): fallback of index_copy_ when not implemented

    * fix(cache): in index_copy_ ensure tensors are on same device

    * [run slow] llama

    * fix(cache): add move of cache_position to same device in SlidingWindowCache

    * Revert "[run slow] llama"

    This reverts commit 02608dd14253ccd464e31c108e0cd94364f0e8b9.

commit a009fbdab32a4b068c24052a4dfe7a7bc0fc89f9
Author: amyeroberts <[email protected]>
Date:   Tue Jul 23 12:23:34 2024 +0100

    Fix typing to be compatible with later py versions (#32155)

commit 3263b3435473cbb5dc66925bc29c1d32b5b8d431
Author: Sanchit Gandhi <[email protected]>
Date:   Tue Jul 23 18:34:30 2024 +0800

    Revert "Incorrect Whisper long-form decoding timestamps " (#32148)

    Revert "Incorrect Whisper long-form decoding timestamps  (#32003)"

    This reverts commit cd48553fc8375e1a28d4d82cfe231dedf6a23af8.

commit 034b47784765e37ecc20f7ad43640f1a2c0094fd
Author: Amit Garg <[email protected]>
Date:   Tue Jul 23 03:33:22 2024 -0700

    Rename Phi-3 rope scaling type (#31436)

    * renamed phi3 rope_scaling type

    * fixed trailing whitespaces

    * fixed test

    * added warning

    * fixed format

commit bab32d6fe932a3372fbd6d5a84e3cacb12a61ae0
Author: Alexandre TL <[email protected]>
Date:   Tue Jul 23 12:32:19 2024 +0200

    Added mamba.py backend (#30139)

    * Update README.md

    * tests: forward ok

    * backward test done

    * done testing

    * removed check. scripts

    * Update README.md

    * added use_mambapy arg

    * fixed typo in warning

    * protected imports w/ mambapy package

    * delete pscan.py + raise rather than assert

    * Update import_utils.py

    * fix whitespaces and unused import

    * trailing whitespace + import block unformatted

    * Update modeling_mamba.py

    * transpose before pscan

    * shape comment

    * ran make style

    * use_mambapy=False by default

    Co-authored-by: Arthur <[email protected]>

    * ran make fix-copies

    ---------

    Co-authored-by: Arthur <[email protected]>

commit 9ced33ca7f909d9ace743dac083daba99c904d46
Author: Merve Noyan <[email protected]>
Date:   Tue Jul 23 13:23:23 2024 +0300

    Fix video batching to videollava (#32139)

    ---------

    Co-authored-by: Merve Noyan <[email protected]>

commit a5b226ce9811aa6b31af0bc9c09c54493a4e67c1
Author: Cyril Vallez <[email protected]>
Date:   Tue Jul 23 12:21:23 2024 +0200

    Fix flash attention speed issue (#32028)

    Add the lru_cache for speed

commit a1844a3209eb7e75582684809203bc189931a90c
Author: Ita Zaporozhets <[email protected]>
Date:   Tue Jul 23 11:45:54 2024 +0200

    gguf conversion add_prefix_space=None for llama3 (#31937)

    * gguf conversion forces add_prefix_space=False for llama3, this is not required and forces from_slow, which fails. changing to None + test

    * typo

    * clean test

commit 2e113422b3504fe6de821bb9911b24273b11aa9c
Author: Joao Gante <[email protected]>
Date:   Tue Jul 23 10:42:55 2024 +0100

    Llama: RoPE refactor (#32135)

    Co-authored-by: amyeroberts <[email protected]>
    Co-authored-by: Arthur <[email protected]>

commit 5a4a76edb7ac6bbc764392e89adc11adda91f3e5
Author: bayllama <[email protected]>
Date:   Tue Jul 23 02:28:44 2024 -0700

    Modify resize_token_embeddings to ensure output type is same as input (#31979)

    * Change resize_token_embeddings to make it return same Class that is passed to it

    * Add explanatory comment as requested in review

    * Add explanatory comments for add resizing function in lxmert

    * Add comment for padding_idx and moving _resize_bias in lxmert to LxmertForPreTraining

    ---------

    Co-authored-by: Prashanth Sateesh <[email protected]>
    Co-authored-by: Prashanth Sateesh <[email protected]>

commit 1535a2c93d325e529dc9a1907f99247fdf8a58e7
Author: Daniel Lok <[email protected]>
Date:   Tue Jul 23 17:26:00 2024 +0800

    Disable quick init for TapasPreTrainedModel (#32149)

    add attribute to model

    Signed-off-by: Daniel Lok <[email protected]>

commit 34b43211d782c00da6fef778dbfaff69bbf3f115
Author: mig-mfreitas <[email protected]>
Date:   Tue Jul 23 10:07:58 2024 +0100

    Add YaRN and Dynamic-YaRN RoPE Scaling Methods (#30910)

    * Add YaRN and Dynamic-YaRN RoPE Scaling Methods

    YaRN (Yet another RoPE extension method) combines the NTK-By-Parts
    Interpolation and Attention Scaling methods, improving upon existing
    RoPE interpolation methods for longer context window sizes.

    Fine-tuned models maintain their original performance across benchmarks
    while enabling efficient extrapolation and transfer learning for
    quicker convergence, especially in compute-limited environments.

    We implement YaRN and Dynamic-YaRN for the following list of models:

     - LLaMA
     - Falcon
     - GPT-NeoX
     - Olmo
     - Persimmon
     - Phi
     - StableLM
     - OpenLLaMA

    New unit tests are added to assert YaRN's correct behavior on both
    short and long sequence inputs.

    For more details, please refer to https://arxiv.org/abs/2309.00071.

    Co-authored-by: Miguel Almeida <[email protected]>

    * Refactor YaRN implementation for LLaMA

    Iterate on YaRN implementation for LLaMA and remove diff from remaining
    models for increased PR modularity.

    This commit includes the following changes:
    - Merge 'yarn_rope_scaling' and 'rope_scaling' dictionaries
    - Remove unnecessary attributes ('extrapolation_factor' and 'finetuned')
      from YaRN classes
    - Inherit 'forward' method in YaRN classes from superclass
    - Rename 'yarn' method to 'compute_yarn_scaling'
    - Extend YaRN tests with further assertions
    - Fix style inconsistencies

    Co-authored-by: Miguel Monte e Freitas <[email protected]>

    * Refactor Tensor Building Logic for YaRN

    - Comply with the the tensor building logic introduced in #30743
    - Add referencing to the optimized Attention Factor equation
    - Remove Dynamic YaRN for a more agile deployment

    Co-authored-by: mig-mfreitas <[email protected]>

    * remove unwanted file

    ---------

    Co-authored-by: Miguel Almeida <[email protected]>
    Co-authored-by: mig-mfreitas <[email protected]>
    Co-authored-by: Joao Gante <[email protected]>

commit 7405c1c77e4637768ea0ad5d27d8a4d8d67bfb19
Author: KonradSzafer <[email protected]>
Date:   Tue Jul 23 10:56:21 2024 +0200

    Add method to retrieve used chat template (#32032)

    encapsulate chat template logic

commit 605f3245dcca34381c35520c35ba0b701ed80d58
Author: Anton Vlasjuk <[email protected]>
Date:   Tue Jul 23 10:11:12 2024 +0200

    Fix mask creations of `GPTNeoX` and `GPT2` (#31944)

    * fix mask creation of gpt2 and gpt_neox caused by me

    * forgot the reshape of masks when shape > 2

    * add tests for gpt neox and gpt2

    * nit on a comment

commit 2782aadae2b0b0c313eac3ee70f84f0335577635
Author: Sanchit Gandhi <[email protected]>
Date:   Tue Jul 23 14:55:16 2024 +0800

    [modelling] remove un-necessary transpose for fa2 attention (#31749)

    * [whisper] remove un-necessary transpose for fa2 attention

    * propagate

commit f83c6f1d02fba5e5ced9357b9c9196c76d937af3
Author: Sanchit Gandhi <[email protected]>
Date:   Tue Jul 23 14:54:38 2024 +0800

    Remove `trust_remote_code` when loading Libri Dummy (#31748)

    * [whisper integration] use parquet dataset for testing

    * propagate to others

    * more propagation

    * last one

commit 3aefb4ec7f957f9561a410eabc6f9d57b2f0384f
Author: Raushan Turganbay <[email protected]>
Date:   Tue Jul 23 10:23:55 2024 +0500

    LLaVaNeXT: pad on right if training (#32134)

    * pad on right if training

    * docs

    * add tests

commit 251a2409c694c29ee28e66c954670c483cf54961
Author: James Thewlis <[email protected]>
Date:   Tue Jul 23 01:12:16 2024 -0400

    Add llama3-llava-next-8b to llava_next conversion script (#31395)

    * Add llama3-llava-next-8b to llava_next conversion script

    Adds support for the lmms-lab/llama3-llava-next-8b model to the
    convert_llava_next_weights_to_hf.py script, along with an example
    prompt generated from the llava_llama_3 conv_template in the LLaVA-NeXT
    repo.

    * Exclude <|begin_of_text|> from prompt example

    This token gets added automatically, so it should not be included in the
    prompt example.

    * Add llava-next-72b and llava-next-110b

    Adds the Qwen-based LLaVA-Next models to the conversion script, along
    with changes to load the models on multiple GPUs for inference.

    * Add llama3 and qwen prompt formats to docs

    * Chat prompt and padding side left for llama3 batched

    * update

    * Update src/transformers/models/llava_next/convert_llava_next_weights_to_hf.py

    Co-authored-by: amyeroberts <[email protected]>

    * Update src/transformers/models/llava_next/convert_llava_next_weights_to_hf.py

    Co-authored-by: amyeroberts <[email protected]>

    * remove code

    * better naming

    ---------

    Co-authored-by: raushan <[email protected]>
    Co-authored-by: Raushan Turganbay <[email protected]>
    Co-authored-by: amyeroberts <[email protected]>

commit 96a074fa7e2c04b904f72d9e827398d4c5f90f25
Author: Marc Sun <[email protected]>
Date:   Mon Jul 22 20:21:59 2024 +0200

    Add new quant method (#32047)

    * Add new quant method

    * update

    * fix multi-device

    * add test

    * add offload

    * style

    * style

    * add simple example

    * initial doc

    * docstring

    * style again

    * works ?

    * better docs

    * switch to non persistant

    * remove print

    * fix init

    * code review

commit bd9dca3b855b5a20ea11097b89c40f34d775f1c7
Author: Arthur <[email protected]>
Date:   Mon Jul 22 19:42:47 2024 +0200

    set warning level to info for special tokens have been added (#32138)

    fixes #7002

commit 817a676bd711f9626e13578068b36ef09cf572dc
Author: amyeroberts <[email protected]>
Date:   Mon Jul 22 18:29:50 2024 +0100

    Don't default to other weights file when use_safetensors=True (#31874)

    * Don't default to other weights file when use_safetensors=True

    * Add tests

    * Update tests/utils/test_modeling_utils.py

    * Add clarifying comments to tests

    * Update tests/utils/test_modeling_utils.py

    * Update tests/utils/test_modeling_utils.py

commit 74d0eb3fedf353bd670aa85ae8fcf4c85f287b5b
Author: Yoni Gottesman <[email protected]>
Date:   Mon Jul 22 20:24:43 2024 +0300

    Return assistant generated tokens mask in apply_chat_template  (#30650)

    return assistant generated tokens mask in apply_chat_template

commit 7987710696803c74ce1b5e7f9dfa055096a6c00e
Author: Bertrand Thia <[email protected]>
Date:   Mon Jul 22 13:08:27 2024 -0400

    [RoBERTa] Minor clarifications to model doc (#31949)

    * minor edits and clarifications

    * address comment

    Co-authored-by: Steven Liu <[email protected]>

    ---------

    Co-authored-by: Steven Liu <[email protected]>

commit 12b6880c81db7742a29ea425dcb9e63b7dbdc449
Author: Sai-Suraj-27 <[email protected]>
Date:   Mon Jul 22 22:16:17 2024 +0530

    fix: Fixed raising `TypeError` instead of `ValueError` for invalid type (#32111)

    * Raised TypeError instead of ValueError for invalid types.

    * Updated formatting using ruff.

    * Retrieved few changes.

    * Retrieved few changes.

    * Updated tests accordingly.

commit d1ec36b94f5ba45fb2423e74074cfedab48cfe73
Author: Woojun Jung <[email protected]>
Date:   Tue Jul 23 00:27:13 2024 +0900

    Update `ko/_toctree.yml` and remove `custom_tools.md` to reflect latest changes (#31969)

    update `ko/_toctree.yml` and remove `custom_tools.md`

commit 7ba028fccb82cbee792b67d596120da8ae9397c9
Author: Matt <[email protected]>
Date:   Mon Jul 22 16:07:29 2024 +0100

    Fix failing test with race condition (#32140)

    * Fix failing test with race condition

    * make fixup

    * monotonic_ns instead of randint

    * uuid4 instead of monotonic_ns

    * Add a finally cleanup step

commit 5a649ff3ecd70599dd0fea7ee430ba47b51a4556
Author: Sanchit Gandhi <[email protected]>
Date:   Mon Jul 22 21:18:48 2024 +0800

    [generate] fix eos/pad id check on mps devices (#31695)

    Co-authored-by: Joao Gante <[email protected]>

commit f2a1e3ca684df624016285266a0ae519e4483be7
Author: Lucain <[email protected]>
Date:   Mon Jul 22 15:14:47 2024 +0200

    Mention model_info.id instead of model_info.modelId (#32106)

commit 0fcfc5ccc968ff5a1a439db04a94f566a0bd1d89
Author: Sai-Suraj-27 <[email protected]>
Date:   Mon Jul 22 18:43:39 2024 +0530

    fix: Replaced deprecated `mktemp()` function (#32123)

    Replaced deprecated mktemp function.

commit c38c55f4fbc0163cc02ef4588fe2ec391171a2f0
Author: Joao Gante <[email protected]>
Date:   Mon Jul 22 14:06:49 2024 +0100

    Generate: store special token tensors under a unique variable name (#31980)

    * rename stuff

    * english; this one shouldn't be changed

    * add a _ to the new var names

    * musicgen

    * derp

commit aa8f86a421e23fe41b6333efc11ea4248e098d83
Author: Brian <[email protected]>
Date:   Mon Jul 22 08:06:22 2024 -0400

    Fix shard order (#32023)

commit b3818805978b411713725a1b7470dc1bda073c29
Author: Aymeric Roucher <[email protected]>
Date:   Mon Jul 22 10:49:57 2024 +0200

    Agents planning (#31702)

    * Allow planning for agents

commit 0fdea8607d7e01eb0e38a1ebeb7feee30a22f0cf
Author: Lucain <[email protected]>
Date:   Fri Jul 19 20:32:39 2024 +0200

    Fix tests after `huggingface_hub` 0.24 (#32054)

    * adapt tests

    * style

    * comment

commit fe008d6ebea1f5770b740991daeefd9322fa434a
Author: Raushan Turganbay <[email protected]>
Date:   Fri Jul 19 19:21:45 2024 +0500

    Chameleon: not supported with fast load (#32091)

    fixes

commit 62aa270f2ab3acca2a58cde8f08400ec49330b03
Author: Zach Mueller <[email protected]>
Date:   Fri Jul 19 08:58:53 2024 -0400

    Disable quick init for deepspeed (#32066)

    Disable via deepspeed

commit 89575b567e061fd87bdd655ba188b6c7a922d54a
Author: Kamil Akesbi <[email protected]>
Date:   Fri Jul 19 13:42:22 2024 +0100

    Support generating with fallback for short form audio in Whisper (#30984)

    * remove is_shortform

    * adapt _retrieve_max_frames_and_seek for short_form

    * return bos token in short and long form

    * add decoder_input_ids to short form audios

    * add eos token for  short form

    * handle short form token_timestamps

    * no need to return scores

    * add is_shortform conditions

    * handle when max_new_tokens is None - short form

    * handle assistant decoding

    * fix

    * handle return_dict_in_generate

    * handle split_by_batch for encoder_attentions attribute

    * handle num_beams>1

    * handle num_return_sequences>1 in generate_with_fallback

    * handle num_return_sequences>1 with return_dict_in_generate=True

    * raise error if max_new_tokens + decoder_inputs_ids > max_target_pos

    * fix

    * apply review suggestions

    * fix

    * Update src/transformers/models/whisper/generation_whisper.py

    Co-authored-by: Sanchit Gandhi <[email protected]>

    * Update src/transformers/models/whisper/generation_whisper.py

    Co-authored-by: Sanchit Gandhi <[email protected]>

    * Update src/transformers/models/whisper/generation_whisper.py

    Co-authored-by: Sanchit Gandhi <[email protected]>

    * fix

    * logits for both short form and long form

    * handle if logits_processor is None

    * test

    * apply review changes to num_return_sequences

    * add _expand_variables_for_generation

    * remove short form commented section

    * update comments

    * uncomment num_beams line in generate_with_fallback

    * update assistant decoding

    * handle return_segment with short form generation

    * up

    * fix output format is_shortform

    * overwrite beam_sample test

    * update _set_return_timestamps

    * apply review suggestions

    * apply review suggestions

    * remove seek_outputs_short_form

    * fix _stack_split_outputs

    * fix stack dim in _stack_split_outputs

    * update tests

    * fix past_key_values + beam tests

    * fix

    * clean _expand_variables_for_generation

    * make style

    * fix slow tests

    * make style

    * max_length condition

    * make style

    * add slow tests for shortform fallback

    * Update src/transformers/models/whisper/generation_whisper.py

    Co-authored-by: Sanchit Gandhi <[email protected]>

    * Update src/transformers/models/whisper/generation_whisper.py

    Co-authored-by: Sanchit Gandhi <[email protected]>

    * apply review changes

    * Update src/transformers/models/whisper/generation_whisper.py

    Co-authored-by: Sanchit Gandhi <[email protected]>

    * up

    * fix slow tests

    * apply review suggestions

    * update test

    * make style

    * small fix

    * fix

    * fix test_new_cache_format

    * fix past_key_values

    * fix

    * make style

    * fix slow tests

    * fix

    ---------

    Co-authored-by: Sanchit Gandhi <[email protected]>

commit 46835ec6aed62e9a73784f1b6a43030afd601e5e
Author: Merve Noyan <[email protected]>
Date:   Fri Jul 19 15:40:40 2024 +0300

    Add image-text-to-text task guide (#31777)

    * Add image-text-to-text task page

    * Update docs/source/en/tasks/image_text_to_text.md

    Co-authored-by: Steven Liu <[email protected]>

    * Update docs/source/en/tasks/image_text_to_text.md

    Co-authored-by: Steven Liu <[email protected]>

    * Update docs/source/en/tasks/image_text_to_text.md

    Co-authored-by: Steven Liu <[email protected]>

    * Update docs/source/en/tasks/image_text_to_text.md

    Co-authored-by: Steven Liu <[email protected]>

    * Update docs/source/en/tasks/image_text_to_text.md

    Co-authored-by: Steven Liu <[email protected]>

    * Update docs/source/en/tasks/image_text_to_text.md

    Co-authored-by: Steven Liu <[email protected]>

    * Update docs/source/en/tasks/image_text_to_text.md

    Co-authored-by: Steven Liu <[email protected]>

    * Update docs/source/en/tasks/image_text_to_text.md

    Co-authored-by: Steven Liu <[email protected]>

    * Update docs/source/en/tasks/image_text_to_text.md

    Co-authored-by: Steven Liu <[email protected]>

    * Update docs/source/en/tasks/image_text_to_text.md

    Co-authored-by: Steven Liu <[email protected]>

    * Update docs/source/en/tasks/image_text_to_text.md

    Co-authored-by: Steven Liu <[email protected]>

    * Address comments

    * Fix heading

    * Update docs/source/en/tasks/image_text_to_text.md

    Co-authored-by: amyeroberts <[email protected]>

    * Update docs/source/en/tasks/image_text_to_text.md

    Co-authored-by: amyeroberts <[email protected]>

    * Update docs/source/en/tasks/image_text_to_text.md

    Co-authored-by: amyeroberts <[email protected]>

    * Update docs/source/en/tasks/image_text_to_text.md

    Co-authored-by: amyeroberts <[email protected]>

    * Update docs/source/en/tasks/image_text_to_text.md

    Co-authored-by: amyeroberts <[email protected]>

    * Update docs/source/en/tasks/image_text_to_text.md

    Co-authored-by: amyeroberts <[email protected]>

    * Address comments

    * Update image_text_to_text.md

    ---------

    Co-authored-by: Steven Liu <[email protected]>
    Co-authored-by: amyeroberts <[email protected]>

commit 4bd8f12972c6ad06e264baa39f17ec9dfa9a5cb2
Author: Merve Noyan <[email protected]>
Date:   Fri Jul 19 14:50:34 2024 +0300

    Fixes to chameleon docs (#32078)

    * Fixes

    * Let's not use auto

commit 566b0f1fbf5feb53a18591ca215a8d1245a790ef
Author: Keith Stevens <[email protected]>
Date:   Fri Jul 19 03:56:45 2024 -0700

    Fix progress callback deepcopy (#32070)

    * Replacing ProgressCallbacks deepcopy with a shallowcopy

    * Using items instead of entries

    * code cleanup for copy in trainer callback

    * Style fix for ProgressCallback

commit e316c5214fe51de0bf8e824245bfd6225c9925aa
Author: Raushan Turganbay <[email protected]>
Date:   Fri Jul 19 15:38:01 2024 +0500

    VideoLLaVa: fix chat format in docs (#32083)

    fix chat format

commit 22f888b3fab3d914882b8f44896a5658712f535c
Author: Joshua Lochner <[email protected]>
Date:   Fri Jul 19 11:19:35 2024 +0200

    [mistral] Fix FA2 attention reshape for Mistral Nemo (#32065)

    * [mistral] Fix FA2 attention reshape

    * [run-slow] mistral

commit cd48553fc8375e1a28d4d82cfe231dedf6a23af8
Author: Kamil Akesbi <[email protected]>
Date:   Fri Jul 19 09:26:38 2024 +0100

    Incorrect Whisper long-form decoding timestamps  (#32003)

    * fix lo form timestamps in decode_batch

    * Update src/transformers/models/whisper/tokenization_whisper.py

    Co-authored-by: Yoach Lacombe <[email protected]>

    * Update src/transformers/models/whisper/tokenization_whisper.py

    Co-authored-by: Yoach Lacombe <[email protected]>

    * add test

    * make style

    * fix copies

    * Update src/transformers/models/whisper/tokenization_whisper_fast.py

    Co-authored-by: amyeroberts <[email protected]>

    * Update src/transformers/models/whisper/tokenization_whisper.py

    Co-authored-by: amyeroberts <[email protected]>

    * Update src/transformers/models/whisper/processing_whisper.py

    Co-authored-by: amyeroberts <[email protected]>

    * Update src/transformers/models/whisper/tokenization_whisper.py

    Co-authored-by: amyeroberts <[email protected]>

    * apply review suggestions

    * fix

    * fix copies

    * fix

    * Update src/transformers/models/whisper/tokenization_whisper_fast.py

    Co-authored-by: amyeroberts <[email protected]>

    * fix-copies

    ---------

    Co-authored-by: Yoach Lacombe <[email protected]>
    Co-authored-by: amyeroberts <[email protected]>

commit 56a7745704261919dd8117e3a8aa4fb43fade30e
Author: NielsRogge <[email protected]>
Date:   Fri Jul 19 10:20:03 2024 +0200

    [Chameleon, Hiera] Improve docs (#32038)

    * Improve docs

    * Fix docs

    * Fix code snippet

commit b873234cb649a24865021f0d598627ce2b24d34a
Author: Raushan Turganbay <[email protected]>
Date:   Fri Jul 19 10:08:56 2024 +0500

    Llava: add default chat templates (#31691)

    * add default chat templates

    * Update src/transformers/models/llava/processing_llava.py

    Co-authored-by: amyeroberts <[email protected]>

    * Update src/transformers/models/llava_next/processing_llava_next.py

    Co-authored-by: amyeroberts <[email protected]>

    * more clear docstring and docs

    * Update docs/source/en/model_doc/llava.md

    Co-authored-by: NielsRogge <[email protected]>

    * Update docs/source/en/model_doc/llava_next.md

    Co-authored-by: NielsRogge <[email protected]>

    * Update docs/source/en/model_doc/vipllava.md

    Co-authored-by: NielsRogge <[email protected]>

    * add tests

    * remove default templates (see #31733)

    * load chat template from another file

    * Update docs/source/en/model_doc/llava_next.md

    Co-authored-by: amyeroberts <[email protected]>

    * revert some changes in docs

    * forgot vipllava

    * chat template file is not temporary hack

    * warn if loading from processor

    * not that file

    * similarly modify `save_pretrained`

    * Update tests/models/llava_next/test_processor_llava_next.py

    Co-authored-by: amyeroberts <[email protected]>

    * Update tests/models/vipllava/test_processor_vipllava.py

    Co-authored-by: amyeroberts <[email protected]>

    * Update docs/source/en/model_doc/vipllava.md

    Co-authored-by: amyeroberts <[email protected]>

    * Update src/transformers/processing_utils.py

    Co-authored-by: amyeroberts <[email protected]>

    * Update src/transformers/processing_utils.py

    Co-authored-by: amyeroberts <[email protected]>

    * Update docs/source/en/model_doc/vipllava.md

    Co-authored-by: amyeroberts <[email protected]>

    * Update docs/source/en/model_doc/llava.md

    Co-authored-by: amyeroberts <[email protected]>

    * Update docs/source/en/model_doc/llava.md

    Co-authored-by: amyeroberts <22614925+amyeroberts@use…
  • Loading branch information
@qubvel
qubvel committed Aug 6, 2024
1 parent 558d9f7 commit f995fdb
Show file tree
Hide file tree
Showing 534 changed files with 12,934 additions and 5,436 deletions.
3 changes: 2 additions & 1 deletion .circleci/config.yml
View file
Original file line number Diff line number Diff line change
Expand Up @@ -142,6 +142,7 @@ jobs:
- run: python utils/custom_init_isort.py --check_only
- run: python utils/sort_auto_mappings.py --check_only
- run: python utils/check_doc_toc.py
- run: python utils/check_docstrings.py --check_all

check_repository_consistency:
working_directory: ~/transformers
Expand Down Expand Up @@ -190,4 +191,4 @@ workflows:
- check_circleci_user
- check_code_quality
- check_repository_consistency
- fetch_all_tests
- fetch_all_tests
2 changes: 1 addition & 1 deletion .github/workflows/self-pr-slow-ci.yml
View file
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@ on:
pull_request:
paths:
- "src/transformers/models/*/modeling_*.py"
- "tests/models/*/test_*.py"
- "tests/**/test_*.py"

concurrency:
group: ${{ github.workflow }}-${{ github.head_ref || github.run_id }}
Expand Down
23 changes: 6 additions & 17 deletions .github/workflows/trufflehog.yml
View file
Original file line number Diff line number Diff line change
Expand Up @@ -10,20 +10,9 @@ jobs:
trufflehog:
runs-on: ubuntu-latest
steps:
- shell: bash
run: |
if [ "${{ github.event_name }}" == "push" ]; then
echo "depth=$(($(jq length <<< '${{ toJson(github.event.commits) }}') + 2))" >> $GITHUB_ENV
echo "branch=${{ github.ref_name }}" >> $GITHUB_ENV
fi
if [ "${{ github.event_name }}" == "pull_request" ]; then
echo "depth=$((${{ github.event.pull_request.commits }}+2))" >> $GITHUB_ENV
echo "branch=${{ github.event.pull_request.head.ref }}" >> $GITHUB_ENV
fi
- name: Checkout code
uses: actions/checkout@v4
with:
ref: ${{env.branch}}
fetch-depth: ${{env.depth}}
- name: Secret Scanning
uses: trufflesecurity/trufflehog@main
- name: Checkout code
uses: actions/checkout@v4
with:
fetch-depth: 0
- name: Secret Scanning
uses: trufflesecurity/trufflehog@main
6 changes: 3 additions & 3 deletions CONTRIBUTING.md
View file
Original file line number Diff line number Diff line change
Expand Up @@ -132,7 +132,7 @@ You will need basic `git` proficiency to contribute to
manual. Type `git --help` in a shell and enjoy! If you prefer books, [Pro
Git](https://git-scm.com/book/en/v2) is a very good reference.

You'll need **[Python 3.8](https://github.com/huggingface/transformers/blob/main/setup.py#L426)** or above to contribute to 🤗 Transformers. Follow the steps below to start contributing:
You'll need **[Python 3.8](https://github.com/huggingface/transformers/blob/main/setup.py#L449)** or above to contribute to 🤗 Transformers. Follow the steps below to start contributing:

1. Fork the [repository](https://github.com/huggingface/transformers) by
clicking on the **[Fork](https://github.com/huggingface/transformers/fork)** button on the repository's page. This creates a copy of the code
Expand Down Expand Up @@ -341,12 +341,12 @@ RUN_SLOW=yes python -m pytest -n auto --dist=loadfile -s -v ./tests/models/my_ne
RUN_SLOW=yes python -m pytest -n auto --dist=loadfile -s -v ./examples/pytorch/text-classification
```

Like the slow tests, there are other environment variables available which not enabled by default during testing:
Like the slow tests, there are other environment variables available which are not enabled by default during testing:
- `RUN_CUSTOM_TOKENIZERS`: Enables tests for custom tokenizers.
- `RUN_PT_FLAX_CROSS_TESTS`: Enables tests for PyTorch + Flax integration.
- `RUN_PT_TF_CROSS_TESTS`: Enables tests for TensorFlow + PyTorch integration.

More environment variables and additional information can be found in the [testing_utils.py](src/transformers/testing_utils.py).
More environment variables and additional information can be found in the [testing_utils.py](https://github.com/huggingface/transformers/blob/main/src/transformers/testing_utils.py).

🤗 Transformers uses `pytest` as a test runner only. It doesn't use any
`pytest`-specific features in the test suite itself.
Expand Down
1 change: 1 addition & 0 deletions Makefile
View file
Original file line number Diff line number Diff line change
Expand Up @@ -56,6 +56,7 @@ quality:
python utils/custom_init_isort.py --check_only
python utils/sort_auto_mappings.py --check_only
python utils/check_doc_toc.py
python utils/check_docstrings.py --check_all


# Format source code automatically and check is there are any problems left that need manual fixing
Expand Down
2 changes: 1 addition & 1 deletion docker/consistency.dockerfile
View file
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@ RUN pip install uv && uv venv && uv pip install --no-cache-dir -U pip setuptools
RUN uv pip install --no-cache-dir --upgrade 'torch' --index-url https://download.pytorch.org/whl/cpu
# tensorflow pin matching setup.py
RUN uv pip install --no-cache-dir "tensorflow-cpu<2.16" "tf-keras<2.16"
RUN uv pip install --no-cache-dir "git+https://github.com/huggingface/transformers.git@${REF}#egg=transformers[flax,quality,vision,testing]"
RUN uv pip install --no-cache-dir "git+https://github.com/huggingface/transformers.git@${REF}#egg=transformers[flax,quality,speech,vision,testing]"
RUN git lfs install

RUN pip uninstall -y transformers
Expand Down
2 changes: 1 addition & 1 deletion docker/transformers-all-latest-gpu/Dockerfile
View file
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@ SHELL ["sh", "-lc"]
# The following `ARG` are mainly used to specify the versions explicitly & directly in this docker file, and not meant
# to be used as arguments for docker build (so far).

ARG PYTORCH='2.3.0'
ARG PYTORCH='2.4.0'
# (not always a valid torch version)
ARG INTEL_TORCH_EXT='2.3.0'
# Example: `cu102`, `cu113`, etc.
Expand Down
2 changes: 1 addition & 1 deletion docker/transformers-pytorch-gpu/Dockerfile
View file
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@ ARG REF=main
RUN git clone https://github.com/huggingface/transformers && cd transformers && git checkout $REF

# If set to nothing, will install the latest version
ARG PYTORCH='2.3.0'
ARG PYTORCH='2.4.0'
ARG TORCH_VISION=''
ARG TORCH_AUDIO=''
# Example: `cu102`, `cu113`, etc.
Expand Down
10 changes: 8 additions & 2 deletions docs/source/en/_toctree.yml
View file
Original file line number Diff line number Diff line change
Expand Up @@ -92,11 +92,15 @@
title: Visual Question Answering
- local: tasks/text-to-speech
title: Text to speech
- local: tasks/image_text_to_text
title: Image-text-to-text
title: Multimodal
- isExpanded: false
sections:
- local: generation_strategies
title: Customize the generation strategy
- local: kv_cache
title: Best Practices for Generation with Cache
title: Generation
- isExpanded: false
sections:
Expand Down Expand Up @@ -155,6 +159,8 @@
title: EETQ
- local: quantization/hqq
title: HQQ
- local: quantization/fbgemm_fp8
title: FBGEMM_FP8
- local: quantization/optimum
title: Optimum
- local: quantization/contribute
Expand Down Expand Up @@ -326,8 +332,6 @@
title: CamemBERT
- local: model_doc/canine
title: CANINE
- local: model_doc/chameleon
title: chameleon
- local: model_doc/codegen
title: CodeGen
- local: model_doc/code_llama
Expand Down Expand Up @@ -760,6 +764,8 @@
title: BridgeTower
- local: model_doc/bros
title: BROS
- local: model_doc/chameleon
title: Chameleon
- local: model_doc/chinese_clip
title: Chinese-CLIP
- local: model_doc/clip
Expand Down
51 changes: 51 additions & 0 deletions docs/source/en/agents.md
View file
Original file line number Diff line number Diff line change
Expand Up @@ -509,3 +509,54 @@ agent = ReactCodeAgent(tools=[search_tool])

agent.run("How many more blocks (also denoted as layers) in BERT base encoder than the encoder from the architecture proposed in Attention is All You Need?")
```

## Gradio interface

You can leverage `gradio.Chatbot`to display your agent's thoughts using `stream_to_gradio`, here is an example:

```py
import gradio as gr
from transformers import (
load_tool,
ReactCodeAgent,
HfEngine,
stream_to_gradio,
)

# Import tool from Hub
image_generation_tool = load_tool("m-ric/text-to-image")

llm_engine = HfEngine("meta-llama/Meta-Llama-3-70B-Instruct")

# Initialize the agent with the image generation tool
agent = ReactCodeAgent(tools=[image_generation_tool], llm_engine=llm_engine)


def interact_with_agent(task):
messages = []
messages.append(gr.ChatMessage(role="user", content=task))
yield messages
for msg in stream_to_gradio(agent, task):
messages.append(msg)
yield messages + [
gr.ChatMessage(role="assistant", content="⏳ Task not finished yet!")
]
yield messages


with gr.Blocks() as demo:
text_input = gr.Textbox(lines=1, label="Chat Message", value="Make me a picture of the Statue of Liberty.")
submit = gr.Button("Run illustrator agent!")
chatbot = gr.Chatbot(
label="Agent",
type="messages",
avatar_images=(
None,
"https://em-content.zobj.net/source/twitter/53/robot-face_1f916.png",
),
)
submit.click(interact_with_agent, [text_input], [chatbot])

if __name__ == "__main__":
demo.launch()
```
19 changes: 1 addition & 18 deletions docs/source/en/chat_templating.md
View file
Original file line number Diff line number Diff line change
Expand Up @@ -580,7 +580,7 @@ default template for that model class is used instead. Let's take a look at the
>>> from transformers import AutoTokenizer
>>> tokenizer = AutoTokenizer.from_pretrained("facebook/blenderbot-400M-distill")

>>> tokenizer.default_chat_template
>>> tokenizer.chat_template
"{% for message in messages %}{% if message['role'] == 'user' %}{{ ' ' }}{% endif %}{{ message['content'] }}{% if not loop.last %}{{ ' ' }}{% endif %}{% endfor %}{{ eos_token }}"
```

Expand Down Expand Up @@ -704,23 +704,6 @@ with other names, pass the name of the template you want to the `chat_template`
We find that this can be a bit confusing for users, though - so if you're writing a template yourself, we recommend
trying to put it all in a single template where possible!

### What are "default" templates?

Before the introduction of chat templates, chat handling was hardcoded at the model class level. For backwards
compatibility, we have retained this class-specific handling as default templates, also set at the class level. If a
model does not have a chat template set, but there is a default template for its model class, the `TextGenerationPipeline`
class and methods like `apply_chat_template` will use the class template instead. You can find out what the default
template for your tokenizer is by checking the `tokenizer.default_chat_template` attribute.

This is something we do purely for backward compatibility reasons, to avoid breaking any existing workflows. Even when
the class template is appropriate for your model, we strongly recommend overriding the default template by
setting the `chat_template` attribute explicitly to make it clear to users that your model has been correctly configured
for chat.

Now that actual chat templates have been adopted more widely, default templates have been deprecated and will be
removed in a future release. We strongly recommend setting the `chat_template` attribute for any tokenizers that
still depend on them!

### What template should I use?

When setting the template for a model that's already been trained for chat, you should ensure that the template
Expand Down
2 changes: 1 addition & 1 deletion docs/source/en/conversations.md
View file
Original file line number Diff line number Diff line change
Expand Up @@ -195,7 +195,7 @@ inputs = {key: tensor.to(model.device) for key, tensor in inputs.items()}
print("Tokenized inputs:\n", inputs)

# 4: Generate text from the model
outputs = model.generate(**inputs, max_new_tokens=512, temperature=0.)
outputs = model.generate(**inputs, max_new_tokens=512, temperature=0.1)
print("Generated tokens:\n", outputs)

# 5: Decode the output back to a string
Expand Down
37 changes: 0 additions & 37 deletions docs/source/en/generation_strategies.md
View file
Original file line number Diff line number Diff line change
Expand Up @@ -174,43 +174,6 @@ An increasing sequence: one, two, three, four, five, six, seven, eight, nine, te
```


## KV Cache Quantization

The `generate()` method supports caching keys and values to enhance efficiency and avoid re-computations. However the key and value
cache can occupy a large portion of memory, becoming a bottleneck for long-context generation, especially for Large Language Models.
Quantizing the cache when using `generate()` can significantly reduce memory requirements at the cost of speed.

KV Cache quantization in `transformers` is largely inspired by the paper [KIVI: A Tuning-Free Asymmetric 2bit Quantization for KV Cache]
(https://arxiv.org/abs/2402.02750) and currently supports `quanto` and `HQQ` as backends. For more information on the inner workings see the paper.

To enable quantization of the key-value cache, one needs to indicate `cache_implementation="quantized"` in the `generation_config`.
Quantization related arguments should be passed to the `generation_config` either as a `dict` or an instance of a [`QuantizedCacheConfig`] class.
One has to indicate which quantization backend to use in the [`QuantizedCacheConfig`], the default is `quanto`.

<Tip warning={true}>

Cache quantization can be detrimental if the context length is short and there is enough GPU VRAM available to run without cache quantization.

</Tip>


```python
>>> import torch
>>> from transformers import AutoTokenizer, AutoModelForCausalLM

>>> tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-2-7b-chat-hf")
>>> model = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-2-7b-chat-hf", torch_dtype=torch.float16).to("cuda:0")
>>> inputs = tokenizer("I like rock music because", return_tensors="pt").to(model.device)

>>> out = model.generate(**inputs, do_sample=False, max_new_tokens=20, cache_implementation="quantized", cache_config={"nbits": 4, "backend": "quanto"})
>>> print(tokenizer.batch_decode(out, skip_special_tokens=True)[0])
I like rock music because it's loud and energetic. It's a great way to express myself and rel

>>> out = model.generate(**inputs, do_sample=False, max_new_tokens=20)
>>> print(tokenizer.batch_decode(out, skip_special_tokens=True)[0])
I like rock music because it's loud and energetic. I like to listen to it when I'm feeling
```

## Watermarking

The `generate()` supports watermarking the generated text by randomly marking a portion of tokens as "green".
Expand Down
18 changes: 18 additions & 0 deletions docs/source/en/internal/generation_utils.md
View file
Original file line number Diff line number Diff line change
Expand Up @@ -386,18 +386,36 @@ A [`Constraint`] can be used to force the generation to include specific tokens
- get_seq_length
- reorder_cache

[[autodoc]] OffloadedCache
- update
- prefetch_layer
- evict_previous_layer

[[autodoc]] StaticCache
- update
- get_seq_length
- reset

[[autodoc]] HybridCache
- update
- reset

[[autodoc]] SlidingWindowCache
- update
- reset

[[autodoc]] EncoderDecoderCache
- get_seq_length
- to_legacy_cache
- from_legacy_cache
- reset
- reorder_cache

[[autodoc]] MambaCache
- update_conv_state
- update_ssm_state
- reset

## Watermark Utils

[[autodoc]] WatermarkDetector
Expand Down
Loading

0 comments on commit f995fdb

Please sign in to comment.