accelerate
torch-native pipeline parallelism for big models
#2345
Merged
Overview
Commits
39
Changes
View On GitHub
Commits
  • Broken version
    muellerzr committed 2 years ago
  • Timing I would expect
    muellerzr committed 2 years ago
  • Working version!
    muellerzr committed 2 years ago
  • Use MethodType
    muellerzr committed 2 years ago
  • working test
    muellerzr committed 2 years ago
  • Tests
    muellerzr committed 2 years ago
  • Use no split module classes explicitly
    muellerzr committed 2 years ago
  • Put split_points in pipelien
    muellerzr committed 2 years ago
  • Store split points in hf_split_points
    muellerzr committed 2 years ago
  • fix case num_process=1
    SunMarc committed 2 years ago
  • Allow for dynamic batch padding (#2352)
    muellerzr committed 2 years ago
  • Rm literal
    muellerzr committed 2 years ago
  • Allow users to pass in max_memory
    muellerzr committed 2 years ago
  • Note about recursion
    muellerzr committed 2 years ago
  • Document, document, document
    muellerzr committed 2 years ago
  • Right import check
    muellerzr committed 2 years ago
  • Merge branch 'main' into pippy-integration-v2
    muellerzr committed 2 years ago
  • Fix bug, add tests to multigpu runners
    muellerzr committed 2 years ago
  • Change default to None
    muellerzr committed 2 years ago
  • Start of docs
    muellerzr committed 2 years ago
  • Try again?
    muellerzr committed 2 years ago
  • Try again x2
    muellerzr committed 2 years ago
  • Trailing comma
    muellerzr committed 2 years ago
  • Move import
    muellerzr committed 2 years ago
  • Clean
    muellerzr committed 2 years ago
  • typehint
    muellerzr committed 2 years ago
  • typo
    muellerzr committed 2 years ago
  • From code review
    muellerzr committed 2 years ago
  • Use num_chunks
    muellerzr committed 2 years ago
  • Update tests/test_utils.py
    muellerzr committed 2 years ago
  • Bad copy/paste
    muellerzr committed 2 years ago
  • hf_split_points
    muellerzr committed 2 years ago
  • Apply suggestions from code review
    muellerzr committed 2 years ago
  • Year
    muellerzr committed 2 years ago
  • Nit
    muellerzr committed 2 years ago
  • better title
    muellerzr committed 2 years ago
  • Rephrase
    muellerzr committed 2 years ago
  • Rephrase
    muellerzr committed 2 years ago
  • Try spacing maybe?
    muellerzr committed 2 years ago
FAQ Terms Privacy Refunds Impressum
Loading