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
|