redmoe-ai-v1 commited on
Commit
3c0bd56
·
1 Parent(s): 21959ae

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -2
README.md CHANGED
@@ -6,6 +6,9 @@ base_model: rednote-hilab/dots.llm1.base
6
  tags:
7
  - chat
8
  library_name: transformers
 
 
 
9
  ---
10
 
11
  # dots1
@@ -77,7 +80,7 @@ print(result)
77
  import torch
78
  from transformers import AutoTokenizer, AutoModelForCausalLM, GenerationConfig
79
 
80
- model_name = "/cpfs/user/taishan/model/rgtjf/dots.llm1.inst"
81
  tokenizer = AutoTokenizer.from_pretrained(model_name)
82
 
83
  model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto", torch_dtype=torch.bfloat16, attn_implementation="eager")
@@ -126,4 +129,4 @@ If you find `dots.llm1` is useful or want to use in your projects, please kindly
126
  journal={arXiv preprint arXiv:TBD},
127
  year={2025}
128
  }
129
- ```
 
6
  tags:
7
  - chat
8
  library_name: transformers
9
+ language:
10
+ - en
11
+ - zh
12
  ---
13
 
14
  # dots1
 
80
  import torch
81
  from transformers import AutoTokenizer, AutoModelForCausalLM, GenerationConfig
82
 
83
+ model_name = "rednote-hilab/dots.llm1.inst"
84
  tokenizer = AutoTokenizer.from_pretrained(model_name)
85
 
86
  model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto", torch_dtype=torch.bfloat16, attn_implementation="eager")
 
129
  journal={arXiv preprint arXiv:TBD},
130
  year={2025}
131
  }
132
+ ```