Hugging Face 简介

1from transformers import pipeline
2
3pipe = pipeline("conversational")
4pipe("Hi! I'm Bob.")

1from transformers import pipeline
2from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
3
4hugging_face_model = "facebook/blenderbot-90M"
5
6tokenizer = AutoTokenizer.from_pretrained(hugging_face_model)
7model = AutoModelForSeq2SeqLM.from_pretrained(hugging_face_model)
8
9pipe = pipeline("conversational", model=model, tokenizer=tokenizer)
10pipe("Hi! I'm Bob.")