Spaces:
Running
on
Zero
Running
on
Zero
Remove conditional generation
Browse files
app.py
CHANGED
@@ -203,6 +203,7 @@ def diffusion_chat(question, eot_weight, mask_weight, max_it, pause_length, shar
|
|
203 |
|
204 |
print('started generation')
|
205 |
prompt = f"User: {question}\nAssistant:"
|
|
|
206 |
input_ids = tokenizer.encode(prompt, add_special_tokens=False)
|
207 |
answer_start = find_answer_start(input_ids, assistant_marker_ids)
|
208 |
if answer_start is None:
|
|
|
203 |
|
204 |
print('started generation')
|
205 |
prompt = f"User: {question}\nAssistant:"
|
206 |
+
prompt = ""
|
207 |
input_ids = tokenizer.encode(prompt, add_special_tokens=False)
|
208 |
answer_start = find_answer_start(input_ids, assistant_marker_ids)
|
209 |
if answer_start is None:
|