Salesforce
/
blip2-flan-t5-xxl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fixed typo in FP16 and 8bit examples (#4) 

- Fixed typo in FP16 and 8bit examples (57a4e5eaaed1e408a61eac5aec8d08db2348847f)


Co-authored-by: Björn Plüster  <bjoernp@users.noreply.huggingface.co>
This commit is contained in:
Niels Rogge 2023-02-27 12:53:53 +00:00 committed by system
parent f16db5558f
commit e5f4303e44
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
README.md
View File

@ -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'