fix initial state bug

dottxt-ai · Jan 8, 2024 · 9e313df · 9e313df
1 parent c11277c
commit 9e313df
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/outlines/fsm/regex.py b/outlines/fsm/regex.py
@@ -558,6 +558,7 @@ def create_fsm_index_end_to_end(
         finals_list.append(final)
 
     all_states = set(fsm_info.transitions.values())
+    all_states.add(fsm_info.initial)
     num_states = max(all_states)
     states_to_token_subsets = numba.typed.List(
         [numba.typed.List.empty_list(nb_int_pair_type) for _ in range(num_states + 1)]