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

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -8
README.md CHANGED
@@ -239,23 +239,21 @@ pipeline_tag: image-text-to-text
239
 
240
  ## 4. Usage
241
 
242
- ### step1. Clone the Repository
243
- First, clone the repository to your local machine:
244
 
245
  ```shell
246
  git clone https://github.com/SkyworkAI/Skywork-R1V.git
247
  cd skywork-r1v/inference
248
  ```
249
- ### step2. Set Up the Environment
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
257
-
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 \
 
239
 
240
  ## 4. Usage
241
 
242
+ ### 1. Clone the Repository
 
243
 
244
  ```shell
245
  git clone https://github.com/SkyworkAI/Skywork-R1V.git
246
  cd skywork-r1v/inference
247
  ```
248
+ ### 2. Set Up the Environment
249
 
250
  ```shell
251
+ conda create -n r1-v python=3.10
252
+ conda activate r1-v
253
+ bash setup.sh
254
  ```
255
 
256
+ ### 3. Run the Inference Script
 
 
257
 
258
  ```shell
259
  CUDA_VISIBLE_DEVICES="0,1" python inference_with_transformers.py \