I tried inferencing with the pretrained checkpoint and the results are just a distorted and low resolution version of the input file :( I also tried training my own model with the same results so it must be something I'm doing wrong