Update README.md
Browse files
README.md
CHANGED
|
@@ -45,7 +45,7 @@ from transformers import Qwen2_5_VLForConditionalGeneration, AutoTokenizer, Auto
|
|
| 45 |
from qwen_vl_utils import process_vision_info
|
| 46 |
|
| 47 |
model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
|
| 48 |
-
"TencentBAC/TBAC-VLR1-3B
|
| 49 |
)
|
| 50 |
|
| 51 |
processor = AutoProcessor.from_pretrained("TencentBAC/TBAC-VLR1-3B")
|
|
|
|
| 45 |
from qwen_vl_utils import process_vision_info
|
| 46 |
|
| 47 |
model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
|
| 48 |
+
"TencentBAC/TBAC-VLR1-3B", torch_dtype="auto", device_map="auto"
|
| 49 |
)
|
| 50 |
|
| 51 |
processor = AutoProcessor.from_pretrained("TencentBAC/TBAC-VLR1-3B")
|