trapoom555
commited on
Commit
•
3abe60e
1
Parent(s):
0ba3a57
Modify readme
Browse files
README.md
CHANGED
@@ -42,7 +42,7 @@ import torch
|
|
42 |
import numpy as np
|
43 |
|
44 |
class PhiSentenceEmbedding:
|
45 |
-
def __init__(self, model_path='
|
46 |
self.tokenizer = AutoTokenizer.from_pretrained(model_path)
|
47 |
self.model = AutoModelForCausalLM.from_pretrained(model_path,
|
48 |
torch_dtype=torch.bfloat16,
|
|
|
42 |
import numpy as np
|
43 |
|
44 |
class PhiSentenceEmbedding:
|
45 |
+
def __init__(self, model_path='microsoft/phi-2', adapter_path=None):
|
46 |
self.tokenizer = AutoTokenizer.from_pretrained(model_path)
|
47 |
self.model = AutoModelForCausalLM.from_pretrained(model_path,
|
48 |
torch_dtype=torch.bfloat16,
|