macadeliccc commited on
Commit
a2415da
1 Parent(s): 9e40fe3

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -7,10 +7,10 @@ import torch
7
  from PIL import Image
8
  from transformers import AutoModel, AutoTokenizer
9
 
10
- model = AutoModel.from_pretrained('openbmb/MiniCPM-Llama3-V-2_5', trust_remote_code=True, torch_dtype=torch.float16)
11
  model = model.to(device='cuda')
12
 
13
- tokenizer = AutoTokenizer.from_pretrained('openbmb/MiniCPM-Llama3-V-2_5', trust_remote_code=True)
14
  model.eval()
15
 
16
  image = Image.open('xx.png').convert('RGB')
 
7
  from PIL import Image
8
  from transformers import AutoModel, AutoTokenizer
9
 
10
+ model = AutoModel.from_pretrained('macadeliccc/ShareGPT-4o-MiniCPM-Llama-3-V-2_5', trust_remote_code=True, torch_dtype=torch.float16)
11
  model = model.to(device='cuda')
12
 
13
+ tokenizer = AutoTokenizer.from_pretrained('macadeliccc/ShareGPT-4o-MiniCPM-Llama-3-V-2_5', trust_remote_code=True)
14
  model.eval()
15
 
16
  image = Image.open('xx.png').convert('RGB')