qingshan777
commited on
Commit
•
102ae2e
1
Parent(s):
5e09d5a
Update README.md
Browse files
README.md
CHANGED
@@ -64,7 +64,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
|
|
64 |
import torch
|
65 |
from PIL import Image
|
66 |
|
67 |
-
checkpoint = "
|
68 |
|
69 |
model = AutoModelForCausalLM.from_pretrained(checkpoint, torch_dtype=torch.float16, device_map='cuda', trust_remote_code=True).eval()
|
70 |
tokenizer = AutoTokenizer.from_pretrained(checkpoint, trust_remote_code=True)
|
|
|
64 |
import torch
|
65 |
from PIL import Image
|
66 |
|
67 |
+
checkpoint = "qihoo360/360VL-8B"
|
68 |
|
69 |
model = AutoModelForCausalLM.from_pretrained(checkpoint, torch_dtype=torch.float16, device_map='cuda', trust_remote_code=True).eval()
|
70 |
tokenizer = AutoTokenizer.from_pretrained(checkpoint, trust_remote_code=True)
|