Coercer commited on
Commit
4ee2b08
·
verified ·
1 Parent(s): b155c46

Update Python_Infer_Utils/Swan.py

Browse files
Files changed (1) hide show
  1. Python_Infer_Utils/Swan.py +1 -1
Python_Infer_Utils/Swan.py CHANGED
@@ -55,7 +55,7 @@ class Eagle:
55
  ):
56
  super().__init__()
57
 
58
- self.embeddings = EmbeddingDatabase(tokenizer, embedding_expected_shape)
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)