1. 11 Dec, 2022 1 commit
  2. 21 Feb, 2022 1 commit
  3. 16 Feb, 2022 1 commit
    • Michael Figurnov's avatar
      Improve support for num_recycle=0. · 2cd61ade
      Michael Figurnov authored
      Previously, setting num_recycle=0 in a pretrained recycling model skipped creating some Linears and LayerNorms. This meant that the the offsets of these modules were not applied leading to a degraded performance in that case.
      
      PiperOrigin-RevId: 429075328
      Change-Id: I9257f859521799f45e2deef3803c249311051225
      2cd61ade
  4. 02 Nov, 2021 1 commit
  5. 30 Jul, 2021 1 commit
  6. 27 Jul, 2021 1 commit
    • Tom Ward's avatar
      Fix import ordering. · 1e216f93
      Tom Ward authored
      PiperOrigin-RevId: 387085679
      Change-Id: I73287fcd0a29e899543b64c596e306195a2f435e
      1e216f93
  7. 22 Jul, 2021 1 commit
  8. 15 Jul, 2021 1 commit