Skip to content

Commit 14764aa

Browse files
Rename LLAMATokenizer to SPieceTokenizer.
1 parent b2c995f commit 14764aa

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

comfy/text_encoders/aura_t5.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
from comfy import sd1_clip
2-
from .llama_tokenizer import LLAMATokenizer
2+
from .spiece_tokenizer import SPieceTokenizer
33
import comfy.text_encoders.t5
44
import os
55

@@ -11,7 +11,7 @@ def __init__(self, device="cpu", layer="last", layer_idx=None, dtype=None):
1111
class PT5XlTokenizer(sd1_clip.SDTokenizer):
1212
def __init__(self, embedding_directory=None):
1313
tokenizer_path = os.path.join(os.path.join(os.path.dirname(os.path.realpath(__file__)), "t5_pile_tokenizer"), "tokenizer.model")
14-
super().__init__(tokenizer_path, pad_with_end=False, embedding_size=2048, embedding_key='pile_t5xl', tokenizer_class=LLAMATokenizer, has_start_token=False, pad_to_max_length=False, max_length=99999999, min_length=256, pad_token=1)
14+
super().__init__(tokenizer_path, pad_with_end=False, embedding_size=2048, embedding_key='pile_t5xl', tokenizer_class=SPieceTokenizer, has_start_token=False, pad_to_max_length=False, max_length=99999999, min_length=256, pad_token=1)
1515

1616
class AuraT5Tokenizer(sd1_clip.SD1Tokenizer):
1717
def __init__(self, embedding_directory=None):
Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,9 @@
11
import os
22

3-
class LLAMATokenizer:
3+
class SPieceTokenizer:
44
@staticmethod
55
def from_pretrained(path):
6-
return LLAMATokenizer(path)
6+
return SPieceTokenizer(path)
77

88
def __init__(self, tokenizer_path):
99
import sentencepiece

0 commit comments

Comments
 (0)