How to run PULSE from 16*16 to 128*128?

Hi, I have a task which requires me to do super-resolution from size 16*16 to 128*128(just like the visual results in PULSE paper). However, I found in this code we can only use the pretrained model which is to do SR task to 1024*1024. Can you help me with this issue? Many thanks!