fix name

dottxt-ai · Dec 15, 2024 · e92ead5 · e92ead5
1 parent 79350f3
commit e92ead5
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/outlines/generate/api.py b/outlines/generate/api.py
@@ -352,11 +352,13 @@ def token_generator() -> Iterator[Union[List[str], str, List[List[str]]]]:
                     ]
 
                     generated_sequences = [
-                        self.format_sequence(
-                            self.strip_stop_sequences(sequence, stop_sequences)
+                        (
+                            self.format_sequence(
+                                self.strip_stop_sequences(sequence, stop_sequences)
+                            )
+                            if stop
+                            else sequence
                         )
-                        if stop
-                        else sequence
                         for sequence, stop in zip(
                             generated_sequences, is_stop_at_reached
                         )
@@ -427,7 +429,7 @@ def __init__(self, model, logits_processor, sampler):
         self.model = model
         self.logits_processor = logits_processor
 
-        self.sampling_params = sampler.samplig_params
+        self.sampling_params = sampler.sampling_params
 
     def prepare_generation_parameters(
         self,