Skip to content

v0.9.1

Choose a tag to compare

@Veedubin Veedubin released this 05 Jul 06:15
· 47 commits to main since this release

v0.9.1 — DeepSpeed CUDA compatibility fix

Fixed

  • DeepSpeed CUDA version mismatch: Auto-set DS_SKIP_CUDA_CHECK=1 to handle CUDA minor version mismatch (e.g., 13.3 vs 13.0). CUDA 13.x is ABI-compatible within the major version, so skipping the check is safe. This sets the env var automatically when --use-deepspeed is enabled.
  • Train-base dependencies: Added deepspeed and lomo-optim to train-base dependencies.

See also

  • Full changelog in CHANGELOG.md