liwenyun
/
CLIP
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge b118307608 into a1d071733d

This commit is contained in:
Jan Philip Göpfert 2023-07-08 02:28:45 -07:00 committed by GitHub
parent a1d071733d b118307608
commit 159081c68c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
clip/clip.py
View File

@ -2,7 +2,7 @@ import hashlib
import os
import urllib
import warnings
from typing import Any, Union, List
from typing import Any, Union, List, Sequence
from pkg_resources import packaging
import torch
@ -202,7 +202,7 @@ def load(name: str, device: Union[str, torch.device] = "cuda" if torch.cuda.is_a
return model, _transform(model.input_resolution.item())
def tokenize(texts: Union[str, List[str]], context_length: int = 77, truncate: bool = False) -> Union[torch.IntTensor, torch.LongTensor]:
def tokenize(texts: Union[str, Sequence[str]], context_length: int = 77, truncate: bool = False) -> Union[torch.IntTensor, torch.LongTensor]:
"""
Returns the tokenized representation of given input string(s)