tensor_parallel
tensor_parallel copied to clipboard
Add more predefined configs
We could use more predefined configs for better user experience. Models to consider:
- [x] bert
- [x] gpt2
- [ ] LLaMa
- [ ] electra
- [ ] something else