Steven-vanb

Please update if you find a way to fix this. I'll try finetuning the model to a different scale by just writing code myself in the meantime