I found the xlarge model on huggingface, but it doesn't work because it requires too much memory. Can I replace it with a smaller progen2 model