Skip to content

Commit

Permalink
Fix #1856 by assigning param to attribute (#1870)
Browse files Browse the repository at this point in the history
  • Loading branch information
MaartenGr authored Mar 25, 2024
1 parent cc1c647 commit 4a522ab
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions bertopic/representation/_textgeneration.py
Original file line number Diff line number Diff line change
Expand Up @@ -91,6 +91,7 @@ def __init__(self,
doc_length: int = None,
tokenizer: Union[str, Callable] = None
):
self.random_state = random_state
set_seed(random_state)
if isinstance(model, str):
self.model = pipeline("text-generation", model=model)
Expand Down

0 comments on commit 4a522ab

Please sign in to comment.