more fixes
Browse files- configuration_phi.py +2 -2
- modeling_phi.py +5 -5
configuration_phi.py
CHANGED
@@ -23,8 +23,8 @@ from transformers.utils import logging
|
|
23 |
logger = logging.get_logger(__name__)
|
24 |
|
25 |
PHI_PRETRAINED_CONFIG_ARCHIVE_MAP = {
|
26 |
-
"microsoft/phi-1": "https://huggingface.co/
|
27 |
-
"microsoft/phi-1_5": "https://huggingface.co/
|
28 |
}
|
29 |
|
30 |
|
|
|
23 |
logger = logging.get_logger(__name__)
|
24 |
|
25 |
PHI_PRETRAINED_CONFIG_ARCHIVE_MAP = {
|
26 |
+
"microsoft/phi-1": "https://huggingface.co/microsoft/phi-1/resolve/main/config.json",
|
27 |
+
"microsoft/phi-1_5": "https://huggingface.co/microsoft/phi-1_5/resolve/main/config.json",
|
28 |
}
|
29 |
|
30 |
|
modeling_phi.py
CHANGED
@@ -52,12 +52,12 @@ if is_flash_attn_2_available():
|
|
52 |
|
53 |
logger = logging.get_logger(__name__)
|
54 |
|
55 |
-
_CHECKPOINT_FOR_DOC = "
|
56 |
_CONFIG_FOR_DOC = "PhiConfig"
|
57 |
|
58 |
PHI_PRETRAINED_MODEL_ARCHIVE_LIST = [
|
59 |
-
"
|
60 |
-
"
|
61 |
# See all Phi models at https://huggingface.co/models?filter=phi
|
62 |
]
|
63 |
|
@@ -978,8 +978,8 @@ class PhiForCausalLM(PhiPreTrainedModel):
|
|
978 |
```python
|
979 |
>>> from transformers import AutoTokenizer, PhiForCausalLM
|
980 |
|
981 |
-
>>> model = PhiForCausalLM.from_pretrained("
|
982 |
-
>>> tokenizer = AutoTokenizer.from_pretrained("
|
983 |
|
984 |
>>> prompt = "This is an example script ."
|
985 |
>>> inputs = tokenizer(prompt, return_tensors="pt")
|
|
|
52 |
|
53 |
logger = logging.get_logger(__name__)
|
54 |
|
55 |
+
_CHECKPOINT_FOR_DOC = "microsoft/phi-1"
|
56 |
_CONFIG_FOR_DOC = "PhiConfig"
|
57 |
|
58 |
PHI_PRETRAINED_MODEL_ARCHIVE_LIST = [
|
59 |
+
"microsoft/phi-1",
|
60 |
+
"microsoft/phi-1_5",
|
61 |
# See all Phi models at https://huggingface.co/models?filter=phi
|
62 |
]
|
63 |
|
|
|
978 |
```python
|
979 |
>>> from transformers import AutoTokenizer, PhiForCausalLM
|
980 |
|
981 |
+
>>> model = PhiForCausalLM.from_pretrained("microsoft/phi-1_5")
|
982 |
+
>>> tokenizer = AutoTokenizer.from_pretrained("microsoft/phi-1_5")
|
983 |
|
984 |
>>> prompt = "This is an example script ."
|
985 |
>>> inputs = tokenizer(prompt, return_tensors="pt")
|