Update src/guidellm/data/deserializers/synthetic.py

Co-authored-by: Samuel Monson <[email protected]>
Signed-off-by: Mark Kurtz <[email protected]>

Author: markurtz

src/guidellm/data/deserializers/synthetic.py

@@ -100,7 +100,7 @@ class SyntheticTextDatasetConfig(StandardBaseModel):
     @model_validator(mode="after")
     def check_prefix_options(self) -> SyntheticTextDatasetConfig:
         prefix_count = self.__pydantic_extra__.get("prefix_count", None)  # type: ignore[attr-defined]
-        prefix_tokens = self.__pydantic_extra__.get("prefix_count", None)  # type: ignore[attr-defined]
+        prefix_tokens = self.__pydantic_extra__.get("prefix_tokens", None)  # type: ignore[attr-defined]
         if prefix_count is not None or prefix_tokens is not None:
             if self.prefix_buckets:
                 raise ValueError(