changing to x-clip
Browse files- src/embedder.py +1 -1
src/embedder.py
CHANGED
@@ -4,7 +4,7 @@ from transformers import XCLIPProcessor, XCLIPModel
|
|
4 |
from decord import VideoReader, cpu
|
5 |
|
6 |
class XCLIPEmbedder:
|
7 |
-
def __init__(self, model_name="microsoft/xclip-
|
8 |
self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
9 |
self.model = XCLIPModel.from_pretrained(model_name).to(self.device)
|
10 |
self.processor = XCLIPProcessor.from_pretrained(model_name)
|
|
|
4 |
from decord import VideoReader, cpu
|
5 |
|
6 |
class XCLIPEmbedder:
|
7 |
+
def __init__(self, model_name="microsoft/xclip-large-patch14"):
|
8 |
self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
9 |
self.model = XCLIPModel.from_pretrained(model_name).to(self.device)
|
10 |
self.processor = XCLIPProcessor.from_pretrained(model_name)
|