Update README.md
Browse files
README.md
CHANGED
|
@@ -62,7 +62,7 @@ import torch
|
|
| 62 |
|
| 63 |
from qwen_vl_utils import process_vision_info
|
| 64 |
from datasets import load_dataset
|
| 65 |
-
from transformers import
|
| 66 |
from gui_actor.constants import chat_template
|
| 67 |
from gui_actor.modeling_qwen25vl import Qwen2_5_VLForConditionalGenerationWithPointer
|
| 68 |
from gui_actor.inference import inference
|
|
@@ -70,7 +70,7 @@ from gui_actor.inference import inference
|
|
| 70 |
|
| 71 |
# load model
|
| 72 |
model_name_or_path = "microsoft/GUI-Actor-3B-Qwen2.5-VL"
|
| 73 |
-
data_processor =
|
| 74 |
tokenizer = data_processor.tokenizer
|
| 75 |
model = Qwen2_5_VLForConditionalGenerationWithPointer.from_pretrained(
|
| 76 |
model_name_or_path,
|
|
|
|
| 62 |
|
| 63 |
from qwen_vl_utils import process_vision_info
|
| 64 |
from datasets import load_dataset
|
| 65 |
+
from transformers import AutoProcessor
|
| 66 |
from gui_actor.constants import chat_template
|
| 67 |
from gui_actor.modeling_qwen25vl import Qwen2_5_VLForConditionalGenerationWithPointer
|
| 68 |
from gui_actor.inference import inference
|
|
|
|
| 70 |
|
| 71 |
# load model
|
| 72 |
model_name_or_path = "microsoft/GUI-Actor-3B-Qwen2.5-VL"
|
| 73 |
+
data_processor = AutoProcessor.from_pretrained(model_name_or_path)
|
| 74 |
tokenizer = data_processor.tokenizer
|
| 75 |
model = Qwen2_5_VLForConditionalGenerationWithPointer.from_pretrained(
|
| 76 |
model_name_or_path,
|