Fix typo from last commit

2024-09-08 11:17:03 +02:00
parent bfbd4e33e1
commit a6a7bf1c34
1 changed files with 1 additions and 1 deletions
@@ -32,7 +32,7 @@ class Loader:
            torch.float32 if self.device == torch.device("cpu") else torch.float16
        )

-        self.model = CLIPModel.from_pretrained(path, torch_dtype=self.dtype).eval()
+        self.pipeline = CLIPModel.from_pretrained(path, torch_dtype=self.dtype).eval()
        self.processor = CLIPProcessor.from_pretrained(path)

        return (self,)