Ruurd commited on
Commit
5213031
·
verified ·
1 Parent(s): 02eb393

Fix noise_start to always start at 1.0

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