Spaces:
Running on Zero

Ruurd commited on
Commit
9cf2e5c
·
verified ·
1 Parent(s): 19c039a

Fix initial noise display

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -166,7 +166,7 @@ def diffusion_chat(question, eot_weight, max_it, pause_length, sharpness, cluste
166
  current_tokens, just_noised_indices = noisify_answer(
167
  input_ids, answer_start, threshold=1.0, eot_weight=eot_weight, clustering=clustering, noise_start = noise_start,
168
  )
169
- yield f"<b>Iteration 0 (initial noise):</b><br>" + tokenizer.convert_tokens_to_string([current_tokens]).replace('\n', '<br>')
170
  pause(1)
171
  last_tokens = []
172
  prev_decoded_tokens = []
 
166
  current_tokens, just_noised_indices = noisify_answer(
167
  input_ids, answer_start, threshold=1.0, eot_weight=eot_weight, clustering=clustering, noise_start = noise_start,
168
  )
169
+ yield f"<b>Iteration 0 (initial noise):</b><br>" + tokenizer.decode(current_tokens, skip_special_tokens=True).replace('\n', '<br>')
170
  pause(1)
171
  last_tokens = []
172
  prev_decoded_tokens = []