zhoukz commited on
Commit
b88e283
·
verified ·
1 Parent(s): 5fed20f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -31,7 +31,7 @@ The [original fp32 model](https://huggingface.co/mispeech/midashenglm-7b-0804-fp
31
  ```python
32
  from transformers import AutoModelForCausalLM, AutoProcessor, AutoTokenizer
33
 
34
- model_id = "mispeech/midashenglm-7b-0804-fp16"
35
  model = AutoModelForCausalLM.from_pretrained(model_id, trust_remote_code=True)
36
  tokenizer = AutoTokenizer.from_pretrained(model_id)
37
  processor = AutoProcessor.from_pretrained(model_id, trust_remote_code=True)
 
31
  ```python
32
  from transformers import AutoModelForCausalLM, AutoProcessor, AutoTokenizer
33
 
34
+ model_id = "mispeech/midashenglm-7b-0804-bf16"
35
  model = AutoModelForCausalLM.from_pretrained(model_id, trust_remote_code=True)
36
  tokenizer = AutoTokenizer.from_pretrained(model_id)
37
  processor = AutoProcessor.from_pretrained(model_id, trust_remote_code=True)