shawn0wang commited on
Commit
194ae74
·
verified ·
1 Parent(s): 15f800f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -1
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
  ---