Update Python_Infer_Utils/Swan.py
Browse files
Python_Infer_Utils/Swan.py
CHANGED
|
@@ -55,7 +55,7 @@ class Eagle:
|
|
| 55 |
):
|
| 56 |
super().__init__()
|
| 57 |
|
| 58 |
-
self.embeddings =
|
| 59 |
|
| 60 |
if isinstance(embedding_dir, str):
|
| 61 |
self.embeddings.add_embedding_dir(embedding_dir)
|
|
|
|
| 55 |
):
|
| 56 |
super().__init__()
|
| 57 |
|
| 58 |
+
self.embeddings = worm(tokenizer, embedding_expected_shape)
|
| 59 |
|
| 60 |
if isinstance(embedding_dir, str):
|
| 61 |
self.embeddings.add_embedding_dir(embedding_dir)
|