1. 05 Jun, 2024 1 commit
    • Yury Sulsky's avatar
      Specify dtype=torch.bool to avoid xla error (#31191) · 66875ac0
      Yury Sulsky authored
      The StoppingCriteriaList allocates is_done without specifying dtype=torch.bool. On XLA this allocates a float tensor and causes a failure on the following line:
      
      is_done = is_done | criteria(input_ids, scores, **kwargs)
      
      by attempting to OR float with bool.
      66875ac0
  2. 04 Jun, 2024 14 commits
  3. 03 Jun, 2024 18 commits
  4. 31 May, 2024 7 commits