Mxode commited on
Commit
7d23211
·
verified ·
1 Parent(s): 824931f

Update README_zh-CN.md

Browse files
Files changed (1) hide show
  1. README_zh-CN.md +3 -3
README_zh-CN.md CHANGED
@@ -1,10 +1,10 @@
1
- # NanoTranslator-immersive_translate-365M
2
 
3
  [English](README.md) | 简体中文
4
 
5
  ## Introduction
6
 
7
- NanoTranslator-immersive_translate-365M 是由 [NanoLM-365M-Base](https://huggingface.co/Mxode/NanoLM-365M-Base) 在 [wmt-19](https://huggingface.co/datasets/wmt/wmt19) 数据集上训练了 600 万数据得来的专门用于**中英双语**的翻译模型。
8
 
9
  此模型遵循[沉浸式翻译](https://immersivetranslate.com/)(Immersive Translate)的 prompt 格式进行训练,可以通过 vllm、lmdeploy 等方式部署为 OpenAI 格式接口,从而完成调用。
10
 
@@ -17,7 +17,7 @@ import torch
17
  from typing import Literal
18
  from transformers import AutoModelForCausalLM, AutoTokenizer
19
 
20
- model_path = 'Mxode/NanoTranslator-immersive_translate-365M'
21
 
22
  model = AutoModelForCausalLM.from_pretrained(model_path).to('cuda:0', torch.bfloat16)
23
  tokenizer = AutoTokenizer.from_pretrained(model_path)
 
1
+ # NanoTranslator-immersive_translate-0.5B
2
 
3
  [English](README.md) | 简体中文
4
 
5
  ## Introduction
6
 
7
+ NanoTranslator-immersive_translate-0.5B 是由 [Qwen2-0.5B-Instruct](https://huggingface.co/Qwen/Qwen2-0.5B-Instruct) 在 [BiST](https://huggingface.co/datasets/Mxode/BiST) 和 [wmt19](https://huggingface.co/datasets/wmt/wmt19) 数据集上训练得到的专门用于**中英双语**的翻译模型。
8
 
9
  此模型遵循[沉浸式翻译](https://immersivetranslate.com/)(Immersive Translate)的 prompt 格式进行训练,可以通过 vllm、lmdeploy 等方式部署为 OpenAI 格式接口,从而完成调用。
10
 
 
17
  from typing import Literal
18
  from transformers import AutoModelForCausalLM, AutoTokenizer
19
 
20
+ model_path = 'Mxode/NanoTranslator-immersive_translate-0.5B'
21
 
22
  model = AutoModelForCausalLM.from_pretrained(model_path).to('cuda:0', torch.bfloat16)
23
  tokenizer = AutoTokenizer.from_pretrained(model_path)