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,
|