Fix None gradient to enable maximum-likelihood training. This fixes issue #59.#91

itaigat merged 2 commits intofacebookresearch:mainfacebookresearch/flow_matching:mainfrom

yiboyang:fix-likelihoodyiboyang/flow_matching:fix-likelihoodCopy head branch name to clipboard