Spaces:
Sleeping
Sleeping
Fix noise_start to always start at 1.0
Browse files
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 =
|
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)
|