What are the adjustment to be done for training larger image?

Short Course Q&A How Diffusion Models Work
1

Existing code only allows images of sizes up to 28x28 which is divisible by 4. I am trying to train the model with images size 150x150. Tried different adjustments in ContextUnet class but was unsuccessful. Please help