doggdad commited on
Commit
bcdd774
·
verified ·
1 Parent(s): d0f4a9f

Update src/preprocess/embedding.py

Browse files
Files changed (1) hide show
  1. src/preprocess/embedding.py +2 -1
src/preprocess/embedding.py CHANGED
@@ -3,7 +3,8 @@ from utils import bt_embeddings
3
  from tqdm import tqdm
4
  from typing import List
5
  from langchain_core.embeddings import Embeddings
6
- from langchain_core.pydantic_v1 import BaseModel
 
7
 
8
  class BridgeTowerEmbeddings(BaseModel,Embeddings):
9
  """ BridgeTower embedding model """
 
3
  from tqdm import tqdm
4
  from typing import List
5
  from langchain_core.embeddings import Embeddings
6
+ #from langchain_core.pydantic_v1 import BaseModel
7
+ from pydantic.v1 import BaseModel
8
 
9
  class BridgeTowerEmbeddings(BaseModel,Embeddings):
10
  """ BridgeTower embedding model """