Merge 45afa5f92c into dcba3cb2e2

2024-09-17 21:22:55 -04:00 · 2024-09-17 21:22:55 -04:00 · c2c981b9c7
parent dcba3cb2e2 45afa5f92c
commit c2c981b9c7
1 changed files with 1 additions and 0 deletions
--- a/clip/clip.py
+++ b/clip/clip.py
@ -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) # set the file pointer to the beginning of the file
            state_dict = torch.load(opened_file, map_location="cpu")

    if not jit: