|
from transformers import AutoModelForSeq2SeqLM, AutoTokenizer, pipeline |
|
|
|
mode_name = 'DDDSSS/translation_en-zh' |
|
device = 'cpu' |
|
model = AutoModelForSeq2SeqLM.from_pretrained(mode_name) |
|
|
|
tokenizer = AutoTokenizer.from_pretrained(mode_name) |
|
translation = pipeline("translation_en_to_zh", model=model, tokenizer=tokenizer, |
|
torch_dtype="float", device_map=True,device=device) |
|
x=["If nothing is detected and there is a config.json file, it’s assumed the library is transformers.","By looking into the presence of files such as *.nemo or *saved_model.pb*, the Hub can determine if a model is from NeMo or Keras."] |
|
re = translation(x, max_length=512) |
|
print('翻译为：' ,re) |
