Code reference instanovo constants diffusion config dataset layers model multinomial_diffusion predict inference beam_search diffusion interfaces knapsack knapsack_beam_search transformer dataset decoding layers model predict train utils convert_to_ipc metrics residues