Safetensors
custom_code
kyusonglee commited on
Commit
9734a55
1 Parent(s): b7bfbfe

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -26,8 +26,8 @@ import requests
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)
 
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)