[rank0]: AssertionError: You can't use same `Accelerator()` instance with multiple models when using DeepSpeed

 How did you solve this `[rank0]: AssertionError: You can't use same `Accelerator()` instance with multiple models when using DeepSpeed`

```

    unet, optimizer, train_dataloader, lr_scheduler,KD_teacher_unet = accelerator.prepare(
        unet, optimizer, train_dataloader, lr_scheduler,KD_teacher_unet
    )
```

unet KD_teacher_unet are two different models