o^tgt is calculated but not used anywhere. Also, how come DDIM in line 7 outputs two results, and in line 8 only one result?