Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Relaxed transformers dependency (#270)
## Summary - Make the `transformers` dependency optional so we only have `torch` and `triton` as required deps, which is helpful if you're not using `transformers` for modeling code. This was also causing installation issues for people using slightly older transformers versions. - If transformers is needed, make it compatible with any 4.x version. The specific model being used should dictate the transformers version compatibility. ## Testing Done `pip install -e .[transformers]` `pip install -e .[dev]` A100-80G-PCIe - Hardware Type: A100-80G-PCIe - [x] run `make test` to ensure correctness - [x] run `make checkstyle` to ensure code style - [x] run `make test-convergence` to ensure convergence
- Loading branch information