mlinmg commited on
Commit
eb14604
1 Parent(s): 7bd6c2d

Update modeling_quasar.py

Browse files
Files changed (1) hide show
  1. 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-4.5B"
59
  _CONFIG_FOR_DOC = "QuasarConfig"
60
 
61
  QUASAR_PRETRAINED_MODEL_ARCHIVE_LIST = [
62
- "AstraMindAI/AstraQuasar-4.5B",
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-4.5B", trust_remote_code=True)
1154
- >>> tokenizer = AutoTokenizer.from_pretrained("AstraMindAI/AstraQuasar-4.5B")
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