Fix input_ids instead of current_tokens for first noise iteration

app.py

@@ -164,7 +164,7 @@ def diffusion_chat(question, eot_weight, max_it, pause_length, sharpness, cluste
 
     ori_input_tokens = input_ids
     current_tokens, just_noised_indices = noisify_answer(
-                current_tokens, answer_start, threshold=1.0, eot_weight=eot_weight, clustering=clustering, noise_start = noise_start,
+                input_ids, answer_start, threshold=1.0, eot_weight=eot_weight, clustering=clustering, noise_start = noise_start,
             )
     last_tokens = []
     prev_decoded_tokens = []