1. 06 Oct, 2022 1 commit
    • Antoine Adam's avatar
      remove numpy dependency · 4e38df05
      Antoine Adam authored
      According to the `setup.py` file, only dependencies are torch and einops. But the `bert_padding.py` file requires `numpy` only to multiply the elements of a `torch.Size` object. This change aims at allowing the use of FlashAttention without numpy.
      4e38df05
  2. 05 Aug, 2022 1 commit
  3. 02 Jun, 2022 1 commit
  4. 29 May, 2022 1 commit
  5. 20 May, 2022 1 commit