Update README.md
Browse files
README.md
CHANGED
|
@@ -44,9 +44,12 @@ import requests
|
|
| 44 |
import torch
|
| 45 |
import transformers
|
| 46 |
from PIL import Image
|
| 47 |
-
from otter import
|
|
|
|
|
|
|
|
|
|
|
|
|
| 48 |
|
| 49 |
-
otter_model = OtterModel.from_pretrained("luodian/otter-9b-hf", device_map=auto)
|
| 50 |
tokenizer = model.text_tokenizer
|
| 51 |
image_processor = transformers.CLIPImageProcessor()
|
| 52 |
demo_image_one = Image.open(
|
|
|
|
| 44 |
import torch
|
| 45 |
import transformers
|
| 46 |
from PIL import Image
|
| 47 |
+
from otter.modeling_otter import OtterForConditionalGeneration
|
| 48 |
+
|
| 49 |
+
model = OtterForConditionalGeneration.from_pretrained(
|
| 50 |
+
"luodian/otter-9b-hf", device_map="auto"
|
| 51 |
+
)
|
| 52 |
|
|
|
|
| 53 |
tokenizer = model.text_tokenizer
|
| 54 |
image_processor = transformers.CLIPImageProcessor()
|
| 55 |
demo_image_one = Image.open(
|