Not work for Swin_T model

Thank you guys for your library. It's a quite convenient tool.

However, when I want to implement simplify with Swin_T model, it seems it does not work.

To illustrate the problem, I've shared a simplified version of the code in the following link. Could you kindly review it and assist me in identifying a solution to rectify this issue? Your support would be greatly appreciated.
https://partage.imt.fr/index.php/s/9jemn7WfkBWS5tx

Thanks a lot.

Bắt đầu bằng cách tải và chạy ví dụ đơn giản hóa được chia sẻ tại https://partage.imt.fr/index.php/s/9jemn7WfkBWS5tx với mô hình Swin_T. Theo dõi nơi quá trình đơn giản hóa thất bại và so sánh đường đi đó với một mô hình hoạt động; được xem là hoàn thành khi trường hợp Swin_T được báo cáo hoàn tất thành công hoặc giới hạn được ghi nhận rõ ràng.

Do mô hình lập chỉ mục viết ra từ nội dung của issue.