Spaces:
Running on Zero

Ruurd commited on
Commit
32b1a1f
·
verified ·
1 Parent(s): 12ed3ea

Remove conditional generation

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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: