diff --git a/Sources/Tokenizers/Tokenizer.swift b/Sources/Tokenizers/Tokenizer.swift index 2701c26f..05443067 100644 --- a/Sources/Tokenizers/Tokenizer.swift +++ b/Sources/Tokenizers/Tokenizer.swift @@ -526,9 +526,10 @@ public class PreTrainedTokenizer: @unchecked Sendable, Tokenizer { let token = NSRegularExpression.escapedPattern(for: $0.content) let prefix = $0.prefix ? #"\s*"# : "" let suffix = $0.suffix ? #"\s*"# : "" + guard $0.prefix || $0.suffix else { return "(?:\(token))" } return "\(prefix)(\(token))\(suffix)" }.joined(separator: "|") - addedTokensRegex = try? NSRegularExpression(pattern: addedTokensRegexString, options: []) + addedTokensRegex = unwrappedAddedTokens.isEmpty ? nil : try? NSRegularExpression(pattern: addedTokensRegexString, options: []) self.specialTokens = specialTokens self.addedTokens = Set(addedTokens.keys)