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

Merge 45afa5f92c into dcba3cb2e2

This commit is contained in:
scott-vsi 2024-09-17 21:22:55 -04:00 committed by GitHub
parent dcba3cb2e2 45afa5f92c
commit c2c981b9c7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 0 deletions
Show Stats 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: if jit:
warnings.warn(f"File {model_path} is not a JIT archive. Loading as a state dict instead") warnings.warn(f"File {model_path} is not a JIT archive. Loading as a state dict instead")
jit = False jit = False
opened_file.seek(0) # set the file pointer to the beginning of the file
state_dict = torch.load(opened_file, map_location="cpu") state_dict = torch.load(opened_file, map_location="cpu")
if not jit: if not jit: