Fork of openpi with fixes