程式碼檔案結構
clip.py
CLIP模組提供了以下方法:
clip.available_models()
返回可用的CLIP模型的名稱。
clip.load(name, device=..., jit=False)
根據clip.available_models()返回的模型名稱,返回模型以及模型所需的TorchVision變換。如有必要,它將下載模型。name引數也可以是本地檢查點檔案的路徑。
可以可選地指定執行模型的裝置,預設情況下,如果有CUDA裝置則使用第一個CUDA裝置,否則使用CPU。當jit為False時,將載入非JIT版本的模型。
clip.tokenize(text: Union[str, List[str]], context_length=77)
返回一個LongTensor,其中包含給定文字輸入的標記化序列。這可以作為模型的輸入使用。