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)
|