Remove answer_start check

app.py

@@ -206,9 +206,9 @@ def diffusion_chat(question, eot_weight, mask_weight, max_it, pause_length, shar
     prompt = ""
     input_ids = tokenizer.encode(prompt, add_special_tokens=False)
     answer_start = find_answer_start(input_ids, assistant_marker_ids)
-    if answer_start is None:
-        yield "Error: Could not find Assistant marker in input."
-        return
+    # if answer_start is None:
+    #     yield "Error: Could not find Assistant marker in input."
+    #     return
     answer_start = 0
     
     if len(input_ids) < 256: