Update github branches in gen.py to 2.01