Make ControlNet SD Training Script `torch.compile` compatible (#6525)
* update: make controlnet script torch compile compatible
Signed-off-by: Suvaditya Mukherjee <suvadityamuk@gmail.com>
* update: correct earlier mistakes for compilation
Signed-off-by: Suvaditya Mukherjee <suvadityamuk@gmail.com>
* update: fix code style issues
Signed-off-by: Suvaditya Mukherjee <suvadityamuk@gmail.com>
---------
Signed-off-by: Suvaditya Mukherjee <suvadityamuk@gmail.com>