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