diff --git a/comfy/text_encoders/spiece_tokenizer.py b/comfy/text_encoders/spiece_tokenizer.py index f42116f5e..2157011d9 100644 --- a/comfy/text_encoders/spiece_tokenizer.py +++ b/comfy/text_encoders/spiece_tokenizer.py @@ -31,7 +31,7 @@ class SPieceTokenizer: return self._vocab def __call__(self, string): - out = self.tokenizer.encode(string) + out = self.tokenizer.encode(string) # pylint: disable=no-member return {"input_ids": out} def serialize_model(self): diff --git a/requirements.txt b/requirements.txt index 024cf8602..7966dacb3 100644 --- a/requirements.txt +++ b/requirements.txt @@ -60,3 +60,4 @@ watchdog PySoundFile networkx>=2.6.3 joblib +jaxtyping