8B model fine-tuning configuration #124

Thanks a lot for sharing more models! I am trying to fine-tune the 8B 512*512 model and when I looked at the infinity.py script there is a function for infinity_2b() but I don't see a function for 8B. Could you please clarify if the current framework enables fine-tuning on 8B weights?

Thank you!