Update modeling_quasar.py
Browse files- modeling_quasar.py +4 -4
modeling_quasar.py
CHANGED
|
@@ -55,11 +55,11 @@ except:
|
|
| 55 |
|
| 56 |
logger = logging.get_logger(__name__)
|
| 57 |
|
| 58 |
-
_CHECKPOINT_FOR_DOC = "AstraMindAI/AstraQuasar-
|
| 59 |
_CONFIG_FOR_DOC = "QuasarConfig"
|
| 60 |
|
| 61 |
QUASAR_PRETRAINED_MODEL_ARCHIVE_LIST = [
|
| 62 |
-
"AstraMindAI/AstraQuasar-
|
| 63 |
]
|
| 64 |
|
| 65 |
|
|
@@ -1150,8 +1150,8 @@ class QuasarForCausalLM(QuasarPreTrainedModel):
|
|
| 1150 |
Example:
|
| 1151 |
```python
|
| 1152 |
>>> from transformers import AutoTokenizer, AutoModelForCausalLM
|
| 1153 |
-
>>> model = AutoModelForCausalLM.from_pretrained("AstraMindAI/AstraQuasar-
|
| 1154 |
-
>>> tokenizer = AutoTokenizer.from_pretrained("AstraMindAI/AstraQuasar-
|
| 1155 |
>>> # you can optionally disable the duplicate trick
|
| 1156 |
>>> # model.model.duplicate_trick = False
|
| 1157 |
>>> # You can also disable the duplicate gradient calculation
|
|
|
|
| 55 |
|
| 56 |
logger = logging.get_logger(__name__)
|
| 57 |
|
| 58 |
+
_CHECKPOINT_FOR_DOC = "AstraMindAI/AstraQuasar-4B"
|
| 59 |
_CONFIG_FOR_DOC = "QuasarConfig"
|
| 60 |
|
| 61 |
QUASAR_PRETRAINED_MODEL_ARCHIVE_LIST = [
|
| 62 |
+
"AstraMindAI/AstraQuasar-4B",
|
| 63 |
]
|
| 64 |
|
| 65 |
|
|
|
|
| 1150 |
Example:
|
| 1151 |
```python
|
| 1152 |
>>> from transformers import AutoTokenizer, AutoModelForCausalLM
|
| 1153 |
+
>>> model = AutoModelForCausalLM.from_pretrained("AstraMindAI/AstraQuasar-4B", trust_remote_code=True)
|
| 1154 |
+
>>> tokenizer = AutoTokenizer.from_pretrained("AstraMindAI/AstraQuasar-4B")
|
| 1155 |
>>> # you can optionally disable the duplicate trick
|
| 1156 |
>>> # model.model.duplicate_trick = False
|
| 1157 |
>>> # You can also disable the duplicate gradient calculation
|