Got exception when import kiwi

[gaohuan2015]

Describe the bug
I had already installed openwiki in a conda env. When I import kiwi, I got this kind of exception
ImportError: cannot import name 'SAVE_STATE_WARNING' from 'torch.optim.lr_scheduler'

To Reproduce
Steps to reproduce the behavior:

1. Conda create -n kiwienv python=3.8
2. pip install openkiwi
3. import kiwi and Run this code
4. ImportError: cannot import name 'SAVE_STATE_WARNING' from 'torch.optim.lr_scheduler'

Expected behavior
The exception will not be raised

Screenshots

Environment (please complete the following information):

• OS: [Windows 10]
• OpenKiwi 2.0
• Python 3.8

Additional context
Add any other context about the problem here.