Update README.md
Browse files
README.md
CHANGED
|
@@ -250,6 +250,7 @@ cd skywork-r1v/inference
|
|
| 250 |
|
| 251 |
```shell
|
| 252 |
pip install -r requirements.txt
|
|
|
|
| 253 |
```
|
| 254 |
|
| 255 |
#### step3. Run the Inference Script
|
|
@@ -257,7 +258,10 @@ pip install -r requirements.txt
|
|
| 257 |
Prepare your images and questions, and update them to inference_with_transformers.py
|
| 258 |
|
| 259 |
```shell
|
| 260 |
-
python inference_with_transformers.py
|
|
|
|
|
|
|
|
|
|
| 261 |
```
|
| 262 |
|
| 263 |
---
|
|
|
|
| 250 |
|
| 251 |
```shell
|
| 252 |
pip install -r requirements.txt
|
| 253 |
+
pip install flash-attn --no-build-isolation
|
| 254 |
```
|
| 255 |
|
| 256 |
#### step3. Run the Inference Script
|
|
|
|
| 258 |
Prepare your images and questions, and update them to inference_with_transformers.py
|
| 259 |
|
| 260 |
```shell
|
| 261 |
+
CUDA_VISIBLE_DEVICES="0,1" python inference_with_transformers.py \
|
| 262 |
+
--model_path path \
|
| 263 |
+
--image_paths image1_path \
|
| 264 |
+
--question "your question"
|
| 265 |
```
|
| 266 |
|
| 267 |
---
|