Update README.md
Browse files
README.md
CHANGED
|
@@ -286,9 +286,7 @@ processor = AutoProcessor.from_pretrained(
|
|
| 286 |
|
| 287 |
# Load the model on the available device(s)
|
| 288 |
model = AutoModelForCausalLM.from_pretrained(
|
| 289 |
-
'jinaai/jina-vlm-v1',
|
| 290 |
-
device_map='auto',
|
| 291 |
-
trust_remote_code=True
|
| 292 |
)
|
| 293 |
|
| 294 |
# You can specify a different model dtype and/or attention implementation
|
|
|
|
| 286 |
|
| 287 |
# Load the model on the available device(s)
|
| 288 |
model = AutoModelForCausalLM.from_pretrained(
|
| 289 |
+
'jinaai/jina-vlm-v1', device_map='auto', trust_remote_code=True
|
|
|
|
|
|
|
| 290 |
)
|
| 291 |
|
| 292 |
# You can specify a different model dtype and/or attention implementation
|