Update README.md
Browse files
README.md
CHANGED
@@ -22,7 +22,7 @@ We introduce OS-Genesis, an interaction-driven pipeline that synthesizes high-qu
|
|
22 |
OS-Genesis-7B-WA is a web action model finetuned from [Qwen2-VL-7B-Instruct](https://huggingface.co/Qwen/Qwen2-VL-7B-Instruct).
|
23 |
|
24 |
### OS-Genesis AC Family Models
|
25 |
-
In the following table, we provide an overview of the OS-Genesis
|
26 |
|
27 |
| Model Name | Base Model | Training Data | HF Link |
|
28 |
| :-------------: | :-------------------------------------------------------------------------------------: | :----------------------------------------------------------------------------: | :---------------------------------------------------------: |
|
@@ -46,7 +46,7 @@ from qwen_vl_utils import process_vision_info
|
|
46 |
|
47 |
# Default: Load the model on the available device(s)
|
48 |
model = Qwen2VLForConditionalGeneration.from_pretrained(
|
49 |
-
"OS-Copilot/OS-Genesis-7B-
|
50 |
)
|
51 |
processor = AutoProcessor.from_pretrained("OS-Copilot/OS-Atlas-Base-7B")
|
52 |
|
|
|
22 |
OS-Genesis-7B-WA is a web action model finetuned from [Qwen2-VL-7B-Instruct](https://huggingface.co/Qwen/Qwen2-VL-7B-Instruct).
|
23 |
|
24 |
### OS-Genesis AC Family Models
|
25 |
+
In the following table, we provide an overview of the OS-Genesis WA Family Models used for evaluating the AndroidControl Benchmark.
|
26 |
|
27 |
| Model Name | Base Model | Training Data | HF Link |
|
28 |
| :-------------: | :-------------------------------------------------------------------------------------: | :----------------------------------------------------------------------------: | :---------------------------------------------------------: |
|
|
|
46 |
|
47 |
# Default: Load the model on the available device(s)
|
48 |
model = Qwen2VLForConditionalGeneration.from_pretrained(
|
49 |
+
"OS-Copilot/OS-Genesis-7B-WA", torch_dtype="auto", device_map="auto"
|
50 |
)
|
51 |
processor = AutoProcessor.from_pretrained("OS-Copilot/OS-Atlas-Base-7B")
|
52 |
|