Spaces:
Running on Zero

Ruurd commited on
Commit
77e752d
·
verified ·
1 Parent(s): 74479ff

Show initial noise

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -166,6 +166,8 @@ 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
  last_tokens = []
170
  prev_decoded_tokens = []
171
 
 
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([tok]).replace('\n', '<br>')
170
+
171
  last_tokens = []
172
  prev_decoded_tokens = []
173