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)
|