llogs
Browse files- pretrained.py +1 -1
pretrained.py
CHANGED
@@ -64,7 +64,7 @@ if __name__ == '__main__':
|
|
64 |
print(f'loading model...')
|
65 |
|
66 |
# model = SentenceTransformer("all-MiniLM-L6-v2", cache_folder='./')
|
67 |
-
model = SentenceTransformer("all-MiniLM-L6-v2", cache_folder='./')
|
68 |
#model = SentenceTransformer("models--sentence-transformers--all-MiniLM-L6-v2/snapshots/1a310852cf8e58d22c5ebff537711d504ad4ad66")
|
69 |
model.max_seq_length = 512
|
70 |
print(f'model max lenght is :{model.max_seq_length}')
|
|
|
64 |
print(f'loading model...')
|
65 |
|
66 |
# model = SentenceTransformer("all-MiniLM-L6-v2", cache_folder='./')
|
67 |
+
model = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2", cache_folder='./')
|
68 |
#model = SentenceTransformer("models--sentence-transformers--all-MiniLM-L6-v2/snapshots/1a310852cf8e58d22c5ebff537711d504ad4ad66")
|
69 |
model.max_seq_length = 512
|
70 |
print(f'model max lenght is :{model.max_seq_length}')
|