luodian
/

OTTER-Image-LLaMA7B-LA-InContext

Model card Files Files and versions

luodian commited on May 1, 2023

Commit

3a24c3e

·

1 Parent(s): 284388a

Update README.md

Files changed (1) hide show

README.md +2 -3

README.md CHANGED Viewed

@@ -44,10 +44,9 @@ import requests
 import torch
 import transformers
 from PIL import Image
-model = OtterForConditionalGeneration.from_pretrained(
-    "luodian/otter-9b-hf", device_map="auto"
-)
 tokenizer = model.text_tokenizer
 image_processor = transformers.CLIPImageProcessor()
 demo_image_one = Image.open(

 import torch
 import transformers
 from PIL import Image
+from otter import OtterModel
+otter_model = OtterModel.from_pretrained("luodian/otter-9b-hf", device_map=auto)
 tokenizer = model.text_tokenizer
 image_processor = transformers.CLIPImageProcessor()
 demo_image_one = Image.open(