From db10723a9ec7f185c7c3c342efbd2f1cc51de433 Mon Sep 17 00:00:00 2001 From: "Feng, Jiqing" Date: Sun, 7 Jan 2024 22:33:25 -0800 Subject: [PATCH] fix pipelines example --- optimum/intel/pipelines/__init__.py | 17 ++++------------- 1 file changed, 4 insertions(+), 13 deletions(-) diff --git a/optimum/intel/pipelines/__init__.py b/optimum/intel/pipelines/__init__.py index 61329264a..3ae08416f 100644 --- a/optimum/intel/pipelines/__init__.py +++ b/optimum/intel/pipelines/__init__.py @@ -236,20 +236,11 @@ def pipeline( Examples: ```python - >>> from transformers import pipeline, AutoModelForTokenClassification, AutoTokenizer + >>> import torch + >>> from optimum.intel.pipelines import pipeline - >>> # Sentiment analysis pipeline - >>> analyzer = pipeline("sentiment-analysis") - - >>> # Question answering pipeline, specifying the checkpoint identifier - >>> oracle = pipeline( - ... "question-answering", model="distilbert-base-cased-distilled-squad", tokenizer="bert-base-cased" - ... ) - - >>> # Named entity recognition pipeline, passing in a specific model and tokenizer - >>> model = AutoModelForTokenClassification.from_pretrained("dbmdz/bert-large-cased-finetuned-conll03-english") - >>> tokenizer = AutoTokenizer.from_pretrained("bert-base-cased") - >>> recognizer = pipeline("ner", model=model, tokenizer=tokenizer) + >>> pipe = pipeline('text-generation', 'gpt2', torch_dtype=torch.bfloat16) + >>> pipe("Describe a real-world application of AI in sustainable energy.") ```""" if model_kwargs is None: model_kwargs = {}