Skip to content

Commit

Permalink
Format changes
Browse files Browse the repository at this point in the history
  • Loading branch information
John Staib Matilla committed Jan 13, 2025
1 parent eff5a2f commit 174bca2
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion modyn/models/tokenizers/gpt2_tokenizer.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@
class GPT2TokenizerTransform(HFTokenizerTransform):
def __init__(self, max_token_length: int = 256):
"""Transformer for GPT-2.
Args:
max_token_length: Maximum token length for GPT-2.
"""
Expand Down
2 changes: 1 addition & 1 deletion modyn/models/tokenizers/hf_tokenizer.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@
class HFTokenizerTransform:
def __init__(self, tokenizer: PreTrainedTokenizer, max_token_length: int) -> None:
"""Parent class for tokenizers based on HuggingFace's Transformers.
Args:
tokenizer: Preloaded tokenizer object.
max_token_length: Maximum length for tokenization.
Expand Down

0 comments on commit 174bca2

Please sign in to comment.