feat: use device from config file when predict

[wolf-joe]

Current behavior: Regardless of the device configuration in the configuration file(configs/prediction/default.yaml), the code will only use the CPU for prediction unless refine is enabled.

Expected behavior: Use the corresponding device for predict based on the 'device' item in the configuration file.

[jf1957]

Current behavior: Regardless of the device configuration in the configuration file(configs/prediction/default.yaml), the code will only use the CPU for prediction unless refine is enabled.

Expected behavior: Use the corresponding device for predict based on the 'device' item in the configuration file.

I would like to kindly ask you some questions regarding training on the CelebA dataset. Specifically, based on LaMa-Fourier, what small modifications or fine-tuning strategies could be applied to slightly improve the final performance metrics?

I am currently working on my undergraduate graduation thesis, and these suggestions would be extremely helpful to my research. I would sincerely appreciate it if you could share some advice or insights.

Thank you very much for your time and help.