Working stable release!

unaidedelf8777 · May 7, 2024 · d9701a6 · d9701a6
1 parent 6009a5a
commit d9701a6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/function_sampler/fsm/fsm_utils/src/lazy_index.rs b/function_sampler/fsm/fsm_utils/src/lazy_index.rs
@@ -104,6 +104,8 @@ impl LazyFSMIndex {
 
 #[pymethods]
 impl LazyFSMIndex {
+
+
     pub fn get_state_map(&self, state: u32) -> Option<BTreeMap<u32, u32>> {
         if let Some(token_map) = self.states_to_token_maps.get(&state) {
             return Some(token_map.clone());

diff --git a/function_sampler/sampler.py b/function_sampler/sampler.py
@@ -333,8 +333,6 @@ def __call__(
                         self.fsm_state = self.fsm.next_state(
                             self.fsm_state, int(last_token)
                         )
-                        if self.fsm_state == -1:
-                            raise Exception("Final state achieved")
                         allowed_tokens = self.fsm.allowed_token_ids(self.fsm_state)
 
                     if allowed_tokens == [-2]: