Fix input_ids instead of current_tokens for first noise iteration 74479ff verified Ruurd commited on Apr 14