Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-11-14 17:18:14 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix CLIPTokenizer skipping underscores

Browse source
This commit is contained in:
TyrianOtter 2024-10-01 14:02:03 -04:00 committed by Pierre Chapuis
parent f89c4f720d
commit 590648ebca
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
src/refiners/foundationals/clip/tokenizer.py
View file

@ -44,7 +44,7 @@ class CLIPTokenizer(fl.Module):
# to get rid of the dependence on the `regex` module. Unicode support could
# potentially be added back by leveraging the `\w` character class.
self.token_pattern = re.compile(
pattern=r"""<\|startoftext\|>|<\|endoftext\|>|'s|'t|'re|'ve|'m|'ll|'d|[a-zA-Z]+|[0-9]|[^\s\w]+""",
pattern=r"""<\|startoftext\|>|<\|endoftext\|>|'s|'t|'re|'ve|'m|'ll|'d|[a-zA-Z]+|[0-9]|(?:[^\s\w]|_)+""",
flags=re.IGNORECASE,
)
self.start_of_text_token_id: int = start_of_text_token_id