1. 10 Jun, 2021 1 commit
  2. 09 Jun, 2021 5 commits
    • Patrick von Platen's avatar
      [Wav2Vec2ForPretraining] Correct checkpoints wav2vec2 & fix tests (#12089) · bc6f51e5
      Patrick von Platen authored
      * fix_torch_device_generate_test
      
      * remove @
      
      * fix tests
      bc6f51e5
    • Stas Bekman's avatar
      rm require_version_examples (#12088) · 61e19198
      Stas Bekman authored
      61e19198
    • Anton Lozhkov's avatar
      Wav2Vec2 Pretraining (#11306) · d472bd7b
      Anton Lozhkov authored
      
      
      * Working quantizer forward
      
      * Working quantizer forward
      
      * Clean up unused model parts, test reproducibility
      
      * Working quantizer forward
      
      * Clean up unused model parts, test reproducibility
      
      * Remove custom outputs from the shared ones
      
      * correct conversion
      
      * correct bug
      
      * add first pretrain script
      
      * save intermediate
      
      * static shapes
      
      * save intermediate
      
      * finish first pretrain script version
      
      * more refactor
      
      * remove wanddb
      
      * refactor more
      
      * improve test
      
      * correct perplexity compute bug
      
      * finish model implementation
      
      * add to docs
      
      * finish docs
      
      * finish pretraining script
      
      * finish pretraining script
      
      * remove wandb
      
      * finish PR for merge
      
      * finish config
      
      * finish
      
      * make deepspeed work
      
      * Apply suggestions from code review
      Co-authored-by: default avatarLysandre Debut <lysandre@huggingface.co>
      Co-authored-by: default avatarSylvain Gugger <35901082+sgugger@users.noreply.github.com>
      
      * apply suggestions
      
      * fix flaky test
      Co-authored-by: default avatarpatrickvonplaten <patrick.v.platen@gmail.com>
      Co-authored-by: default avatarLysandre Debut <lysandre@huggingface.co>
      Co-authored-by: default avatarSylvain Gugger <35901082+sgugger@users.noreply.github.com>
      d472bd7b
    • Stas Bekman's avatar
      [test] support more than 2 gpus (#12074) · b1a8aa94
      Stas Bekman authored
      * support more than 2 gpus
      
      * style
      b1a8aa94
    • NielsRogge's avatar
      Add DETR (#11653) · d3eacbb8
      NielsRogge authored
      
      
      * Squash all commits of modeling_detr_v7 branch into one
      
      * Improve docs
      
      * Fix tests
      
      * Style
      
      * Improve docs some more and fix most tests
      
      * Fix slow tests of ViT, DeiT and DETR
      
      * Improve replacement of batch norm
      
      * Restructure timm backbone forward
      
      * Make DetrForSegmentation support any timm backbone
      
      * Fix name of output
      
      * Address most comments by @LysandreJik
      
      * Give better names for variables
      
      * Conditional imports + timm in setup.py
      
      * Address additional comments by @sgugger
      
      * Make style, add require_timm and require_vision to tests茅
      
      * Remove train_backbone attribute of DetrConfig, add methods to freeze/unfreeze backbone
      
      * Add png files to fixtures
      
      * Fix type hint
      
      * Add timm to workflows
      
      * Add `BatchNorm2d` to the weight initialization
      
      * Fix retain_grad test
      
      * Replace model checkpoints by Facebook namespace
      
      * Fix name of checkpoint in test
      
      * Add user-friendly message when scipy is not available
      
      * Address most comments by @patrickvonplaten
      
      * Remove return_intermediate_layers attribute of DetrConfig and simplify Joiner
      
      * Better initialization
      
      * Scipy is necessary to get sklearn metrics
      
      * Rename TimmBackbone to DetrTimmConvEncoder and rename DetrJoiner to DetrConvModel
      
      * Make style
      
      * Improve docs and add 2 community notebooks
      Co-authored-by: default avatarLysandre <lysandre.debut@reseau.eseo.fr>
      d3eacbb8
  3. 08 Jun, 2021 6 commits
  4. 07 Jun, 2021 2 commits
  5. 04 Jun, 2021 1 commit
  6. 02 Jun, 2021 4 commits
  7. 01 Jun, 2021 6 commits
  8. 31 May, 2021 1 commit
  9. 28 May, 2021 2 commits
  10. 27 May, 2021 1 commit
    • Nicolas Patry's avatar
      Adding new argument `max_new_tokens` for generate. (#11476) · 80d712fa
      Nicolas Patry authored
      * Adding new argument `max_new_tokens` for generate.
      
      This is a proposal to add a new argument `max_new_tokens` to `generate`.
      This include a `MaxNewTokensCriteria` that enables callers that don't
      know about the token length ahead (like pipelines callers) to manage
      more easily the length of their generated output.
      
      * Adding a test for the user warning when both`max_length` and
      `max_new_tokens` are used together.
      
      * Removed redundant `no_grad`.
      80d712fa
  11. 26 May, 2021 3 commits
    • Patrick von Platen's avatar
      Flax Generate (#11777) · 996a315e
      Patrick von Platen authored
      
      
      * fix_torch_device_generate_test
      
      * remove @
      
      * add
      
      * indexing
      
      * correct a couple of tests
      
      * fix tests
      
      * add logits processor
      
      * finish top_k, top_p, temp
      
      * add docs
      
      * correct flax prng key default
      
      * improve generate
      
      * add generation docs
      
      * add docs
      
      * make style
      
      * revert model outputs change
      
      * make style
      
      * correct typo
      
      * fix tests
      
      * fix slow test
      
      * add raise
      
      * finish generation
      Co-authored-by: default avatarPatrick von Platen <patrick@huggingface.co>
      996a315e
    • Patrick von Platen's avatar
      [Flax] Allow dataclasses to be jitted (#11886) · d5a72b6e
      Patrick von Platen authored
      * fix_torch_device_generate_test
      
      * remove @
      
      * change dataclasses to flax ones
      
      * fix typo
      
      * fix jitted tests
      
      * fix bert & electra
      d5a72b6e
    • Daniel Stancl's avatar
      Fix usage of head masks by TF encoder-decoder models' `generate()` function (#11775) · 0b933584
      Daniel Stancl authored
      * Fix Bart
      
      * Fix Blenderbot{,_small}
      
      * Fix LED
      
      * Fix Marian
      
      * Fix MBart
      
      * Fix Pegasus
      
      * Fix T5
      
      * Add test for generation with head_mask
      
      * Add a common TF test
      
      * Override a test for the LED model as head masking is not yet properly implemented
      
      * Remove all head_masks from input preparation for LED
      
      * Drop masking for T5 as it needs a bit of refactor
      0b933584
  12. 25 May, 2021 5 commits
  13. 24 May, 2021 1 commit
  14. 21 May, 2021 2 commits