Update README.md
Browse files
README.md
CHANGED
@@ -48,13 +48,13 @@ from transformers import Qwen2_5_VLForConditionalGeneration, AutoTokenizer, Auto
|
|
48 |
from qwen_vl_utils import process_vision_info
|
49 |
# default: Load the model on the available device(s)
|
50 |
model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
|
51 |
-
"
|
52 |
torch_dtype=torch.bfloat16,
|
53 |
attn_implementation="flash_attention_2",
|
54 |
device_map="auto",
|
55 |
)
|
56 |
# default processer
|
57 |
-
processor = AutoProcessor.from_pretrained("
|
58 |
# Example input
|
59 |
messages = [
|
60 |
{
|
|
|
48 |
from qwen_vl_utils import process_vision_info
|
49 |
# default: Load the model on the available device(s)
|
50 |
model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
|
51 |
+
"Fintor/Qwen2.5-VL-7B-Instruct-Finetuned-Os-Atlas",
|
52 |
torch_dtype=torch.bfloat16,
|
53 |
attn_implementation="flash_attention_2",
|
54 |
device_map="auto",
|
55 |
)
|
56 |
# default processer
|
57 |
+
processor = AutoProcessor.from_pretrained("Fintor/Qwen2.5-VL-7B-Instruct-Finetuned-Os-Atlas")
|
58 |
# Example input
|
59 |
messages = [
|
60 |
{
|