zR commited on
Update README.md
Browse files
README.md
CHANGED
|
@@ -35,8 +35,10 @@ device = "cuda"
|
|
| 35 |
tokenizer = AutoTokenizer.from_pretrained("THUDM/glm-4v-9b", trust_remote_code=True)
|
| 36 |
|
| 37 |
query = '描述这张图片'
|
|
|
|
| 38 |
image = Image.open("your image").convert('RGB')
|
| 39 |
-
|
|
|
|
| 40 |
add_generation_prompt=True, tokenize=True, return_tensors="pt",
|
| 41 |
return_dict=True) # chat mode
|
| 42 |
|
|
|
|
| 35 |
tokenizer = AutoTokenizer.from_pretrained("THUDM/glm-4v-9b", trust_remote_code=True)
|
| 36 |
|
| 37 |
query = '描述这张图片'
|
| 38 |
+
|
| 39 |
image = Image.open("your image").convert('RGB')
|
| 40 |
+
|
| 41 |
+
inputs = tokenizer.apply_chat_template([{"role": "user", "image": image, "content": query}],
|
| 42 |
add_generation_prompt=True, tokenize=True, return_tensors="pt",
|
| 43 |
return_dict=True) # chat mode
|
| 44 |
|