Spaces:

Ruurd
/

tini

Running on Zero

Ruurd commited on May 6

Commit

4d3e9fe

verified ·

1 Parent(s): 32b1a1f

Set answer_start to 0

Files changed (1) hide show

app.py CHANGED Viewed

@@ -209,7 +209,8 @@ def diffusion_chat(question, eot_weight, mask_weight, max_it, pause_length, shar
     if answer_start is None:
         yield "Error: Could not find Assistant marker in input."
         return
     if len(input_ids) < 256:
         input_ids += [pad_token] * (256 - len(input_ids))
     else:

     if answer_start is None:
         yield "Error: Could not find Assistant marker in input."
         return
+    answer_start = 0
     if len(input_ids) < 256:
         input_ids += [pad_token] * (256 - len(input_ids))
     else: