How to load COMPVIS_CHECKPOINT to UnlearnAttack

[YuanXun2024]

Hi,

Thanks for releasing the code. I noted that you provided results that use UnlearnAttack to attack the unlearned diffusion models. But many of the parameters are saved as COMPVIS_CHECKPOINT while UnlearnAttack requires Diffuser checkpoint.

Could you provide the code for converting COMPVIS_CHECKPOINT to Diffuser checkpoint.

Thanks a lot.