Commit ·
2ba12a5
1
Parent(s): 862fc47
Update README.md
Browse files
README.md
CHANGED
|
@@ -27,7 +27,7 @@ If you find this model helpful, please *like* this model and star us on https://
|
|
| 27 |
### 模型使用
|
| 28 |
``` python
|
| 29 |
import torch
|
| 30 |
-
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 31 |
model_dir = '/path/to_finetuned_model/'
|
| 32 |
img_path = 'you_image_path'
|
| 33 |
tokenizer = AutoTokenizer.from_pretrained(model_dir, trust_remote_code=True)
|
|
|
|
| 27 |
### 模型使用
|
| 28 |
``` python
|
| 29 |
import torch
|
| 30 |
+
from transformers import AutoModelForCausalLM, AutoTokenizer, GenerationConfig
|
| 31 |
model_dir = '/path/to_finetuned_model/'
|
| 32 |
img_path = 'you_image_path'
|
| 33 |
tokenizer = AutoTokenizer.from_pretrained(model_dir, trust_remote_code=True)
|