only train and infer stage 2 model

[tzayuan]

Hi, @0x3f3f3f3fun I want to train & inference only stage 2 model(stage 1 will implement alone by another model and generate restorationed image), if you could provide me a command structures?(how to only fine-tune stage 2 model and save it, and how to inference only stage 2 model) Thanks!

BRs, tzayuan