1. 13 Jun, 2022 1 commit
  2. 09 Jun, 2022 2 commits
    • Nicolas Patry's avatar
      Running a pipeline of `float16`. (#17637) · c38f4e1f
      Nicolas Patry authored
      When we're preparing the tensors for CPU for postprocessing, we need
      to upgrade the `float16` to `float32` since CPUs don't have instructions
      for `[b]float16`.
      c38f4e1f
    • Nicolas Patry's avatar
      Adding `top_k` argument to `text-classification` pipeline. (#17606) · 2351729f
      Nicolas Patry authored
      * Adding `top_k` and `sort` arguments to `text-classification` pipeline.
      
      - Deprecate `return_all_scores` as `top_k` is more uniform with other
        pipelines, and a superset of what `return_all_scores` can do.
        BC is maintained though.
        `return_all_scores=True` -> `top_k=None`
        `return_all_scores=False` -> `top_k=1`
      
      - Using `top_k` will imply sorting the results, but using no argument
        will keep the results unsorted for backward compatibility.
      
      * Remove `sort`.
      
      * Fixing the test.
      
      * Remove bad doc.
      2351729f
  3. 19 May, 2022 2 commits
  4. 18 May, 2022 1 commit
  5. 12 May, 2022 1 commit
  6. 10 May, 2022 1 commit
  7. 05 May, 2022 1 commit
  8. 21 Apr, 2022 1 commit
  9. 20 Apr, 2022 1 commit
  10. 14 Apr, 2022 1 commit
  11. 12 Apr, 2022 1 commit
    • Nicolas Patry's avatar
      Change the chunk_iter function to handle (#16730) · a192f61e
      Nicolas Patry authored
      * Change the chunk_iter function to handle
      
      the subtle cases where the last chunk gets ignored since all the
      data is in the `left_strided` data.
      
      We need to remove the right striding on the previous item.
      
      * Remove commented line.
      a192f61e
  12. 18 Mar, 2022 1 commit
  13. 09 Mar, 2022 1 commit
  14. 04 Mar, 2022 2 commits
  15. 03 Mar, 2022 2 commits
  16. 02 Mar, 2022 1 commit
  17. 28 Feb, 2022 1 commit
  18. 25 Feb, 2022 2 commits
  19. 23 Feb, 2022 1 commit