FlagAI
FlagAI copied to clipboard
fix Update grads.py Torch does not support torch._six anymore
Torch does not support torch._six anymore and it has been removed. # Refer - pytorch/pytorch#94709
name: Pull Request title: '[PR]' assignees: 'BAAI-OpenPlatform,ftgreat'
Description
Please describe here what the PR does.
Checklist
- [ ] bug fixed
- [ ] new feature provided
- [ ] documentation updated
- [ ] tests added