Generating Dialogue Issues about color4dial HOT 5 OPEN

/usr/local/lib/python3.10/dist-packages/transformers/models/bart/modeling_bart.py:549: FutureWarning: The class `PretrainedBartModel` has been depreciated, please use `BartPreTrainedModel` instead.
  warnings.warn(
Traceback (most recent call last):
  File "/content/Color4Dial/main_dialog.py", line 22, in <module>
    from train.trainer_dialog import IgniteTrainer
  File "/content/Color4Dial/train/trainer_dialog.py", line 9, in <module>
    from ignite.engine import Engine, Events
  File "/usr/local/lib/python3.10/dist-packages/ignite/__init__.py", line 1, in <module>
    import ignite.engine
  File "/usr/local/lib/python3.10/dist-packages/ignite/engine/__init__.py", line 3, in <module>
    from ignite.engine.engine import Engine, State, Events
  File "/usr/local/lib/python3.10/dist-packages/ignite/engine/engine.py", line 9, in <module>
    from ignite._utils import _to_hours_mins_secs
  File "/usr/local/lib/python3.10/dist-packages/ignite/_utils.py", line 3, in <module>
    from ignite.utils import convert_tensor, apply_to_tensor, apply_to_type, to_onehot
  File "/usr/local/lib/python3.10/dist-packages/ignite/utils.py", line 4, in <module>
    from torch._six import string_classes
ModuleNotFoundError: No module named 'torch._six'