# Transfer Learning on DMs for Images

To run, use
```
python scripts/inference_test.py
```

In the entry file above, we have the customize scheduler and customized pipeline for edition.

## Reward Function
For reward functions, run this file `reward_models/aesthetic_score/reward_model.py`. The example of computing gradient of preprocessed image to the reward function is also contained in the file.