Training Error For Erase 100 Art Styles

[VMS-6511]

Hi,

I believe there is an error in the repo such that it is not possible to replicate the erase 100 art experiments. Currently, in Line 64 of cfr_lora_training.py there is an if statement to determine if the masks are None are not depending on the task. Meanwhile, the YAML file for erase_art_100 has set use_gsam_mask: true. This results in an error. I am wondering if the artistic style experiments were run with or without these masks? Thanks!

Best,
Vinith

[Shilin-LU]

Hi, thank you for finding the error, the artistic style experiments were run without masks