kyusonglee
commited on
Commit
•
9734a55
1
Parent(s):
b7bfbfe
Update README.md
Browse files
README.md
CHANGED
@@ -26,8 +26,8 @@ import requests
|
|
26 |
import torch
|
27 |
from transformers import TextStreamer
|
28 |
|
29 |
-
model = AutoModel.from_pretrained("
|
30 |
-
processor = AutoProcessor.from_pretrained("
|
31 |
|
32 |
url = "https://www.ilankelman.org/stopsigns/australia.jpg"
|
33 |
image = Image.open(requests.get(url, stream=True).raw)
|
|
|
26 |
import torch
|
27 |
from transformers import TextStreamer
|
28 |
|
29 |
+
model = AutoModel.from_pretrained("omlab/omchat-v2.0-13B-single-beta_hf",trust_remote_code=True, torch_dtype=torch.float16).cuda().eval()
|
30 |
+
processor = AutoProcessor.from_pretrained("omlab/omchat-v2.0-13B-single-beta_hf", trust_remote_code=True)
|
31 |
|
32 |
url = "https://www.ilankelman.org/stopsigns/australia.jpg"
|
33 |
image = Image.open(requests.get(url, stream=True).raw)
|