From e5f4303e447c0863f903648f3fa88ca536c4328c Mon Sep 17 00:00:00 2001 From: Niels Rogge Date: Mon, 27 Feb 2023 12:53:53 +0000 Subject: [PATCH] Fixed typo in FP16 and 8bit examples (#4) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - Fixed typo in FP16 and 8bit examples (57a4e5eaaed1e408a61eac5aec8d08db2348847f) Co-authored-by: Björn Plüster --- README.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 4a180ff..d10c0b0 100644 --- a/README.md +++ b/README.md @@ -108,7 +108,7 @@ import requests from PIL import Image from transformers import Blip2Processor, Blip2ForConditionalGeneration -processor = Bli2pProcessor.from_pretrained("Salesforce/blip2-flan-t5-xxl") +processor = Blip2Processor.from_pretrained("Salesforce/blip2-flan-t5-xxl") model = Blip2ForConditionalGeneration.from_pretrained("Salesforce/blip2-flan-t5-xxl", torch_dtype=torch.float16, device_map="auto") img_url = 'https://storage.googleapis.com/sfr-vision-language-research/BLIP/demo.jpg' @@ -134,7 +134,7 @@ import requests from PIL import Image from transformers import Blip2Processor, Blip2ForConditionalGeneration -processor = Bli2pProcessor.from_pretrained("Salesforce/blip2-flan-t5-xxl") +processor = Blip2Processor.from_pretrained("Salesforce/blip2-flan-t5-xxl") model = Blip2ForConditionalGeneration.from_pretrained("Salesforce/blip2-flan-t5-xxl", load_in_8bit=True, device_map="auto") img_url = 'https://storage.googleapis.com/sfr-vision-language-research/BLIP/demo.jpg'