From f75de602df20a8961c39b49f9583d7795bf34e2c Mon Sep 17 00:00:00 2001 From: AN Long Date: Wed, 12 Jul 2023 06:26:40 +0800 Subject: [PATCH] fix typo in stable diffusion example (#1219) --- examples/stable_diffusion.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/stable_diffusion.py b/examples/stable_diffusion.py index ad250a2011..ceef68eec1 100644 --- a/examples/stable_diffusion.py +++ b/examples/stable_diffusion.py @@ -519,7 +519,7 @@ class ClipTokenizer: self.encoder = dict(zip(vocab, range(len(vocab)))) self.bpe_ranks = dict(zip(merges, range(len(merges)))) self.cache = {'<|startoftext|>': '<|startoftext|>', '<|endoftext|>': '<|endoftext|>'} - self.pat = self.pat = re.compile(r"""<\|startoftext\|>|<\|endoftext\|>|'s|'t|'re|'ve|'m|'ll|'d|[^\s]+""", re.IGNORECASE) + self.pat = re.compile(r"""<\|startoftext\|>|<\|endoftext\|>|'s|'t|'re|'ve|'m|'ll|'d|[^\s]+""", re.IGNORECASE) def bpe(self, token): if token in self.cache: