openai/CLIP
1
0
Fork 0
mirror of https://github.com/openai/CLIP.git synced 2026-03-14 22:07:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixes #396

Browse Source
This commit is contained in:
Jong Wook Kim
2023-10-16 15:47:21 -07:00
committed by GitHub
parent a1d071733d
commit 551aca3475
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
clip/clip.py
View File

@@ -133,6 +133,7 @@ def load(name: str, device: Union[str, torch.device] = "cuda" if torch.cuda.is_a
if jit:
warnings.warn(f"File {model_path} is not a JIT archive. Loading as a state dict instead")
jit = False
opened_file.seek(0)
state_dict = torch.load(opened_file, map_location="cpu")
if not jit: