Clip模型使用

seekwhale13發表於2024-11-29

image

程式碼檔案結構

image

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,其中包含給定文字輸入的標記化序列。這可以作為模型的輸入使用。

model.py

相關文章