Update README.md
Browse files
README.md
CHANGED
|
@@ -125,7 +125,7 @@ def load_image(image_file, input_size=448, max_num=12):
|
|
| 125 |
return pixel_values
|
| 126 |
|
| 127 |
# If you want to load a model using multiple GPUs, please refer to the `Multiple GPUs` section.
|
| 128 |
-
path = 'OS-Copilot/OS-Genesis-8B-
|
| 129 |
model = AutoModel.from_pretrained(
|
| 130 |
path,
|
| 131 |
torch_dtype=torch.bfloat16,
|
|
|
|
| 125 |
return pixel_values
|
| 126 |
|
| 127 |
# If you want to load a model using multiple GPUs, please refer to the `Multiple GPUs` section.
|
| 128 |
+
path = 'OS-Copilot/OS-Genesis-8B-WA'
|
| 129 |
model = AutoModel.from_pretrained(
|
| 130 |
path,
|
| 131 |
torch_dtype=torch.bfloat16,
|