Using specific inputs and outputs with dream.train_step

[Ednaordinary]

Hi There!

I was wondering if you can input a specific file to the model (via dream.train_step()) each iteration, so effects like cropping can be applied between each iteration before generating?
For example:
Uses BigSleep+CLIP to do an iteration
Applies a crop to the image
Replaces the generated image with the cropped image or passes the path to the cropped image to dream.train_step(), so the input is slightly zoomed in
Loops that 400 times