Skip to content

Commit

Permalink
refine readme
Browse files Browse the repository at this point in the history
  • Loading branch information
lixiang007666 committed Jul 26, 2024
1 parent 3bc6425 commit 2808806
Showing 1 changed file with 19 additions and 4 deletions.
23 changes: 19 additions & 4 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -52,25 +52,40 @@ Evaluating the use of all generative models is divided into two steps, taking th
```
# Original Pytorch generates reference images.
python3 models/kolors/text_to_image_kolors_quality_benchmark.py --dataset coco --csv-file resources/MS-COCO_val2014_30k_captions.csv --output-dir /path/to/your/kolors_torch_coco
python3 models/kolors/text_to_image_kolors_quality_benchmark.py \
--dataset coco \
--csv-file resources/MS-COCO_val2014_30k_captions.csv \
--output-dir /path/to/your/kolors_torch_coco
```
```
# Accelerate using onediff's oneflow backend.
python3 models/kolors/text_to_image_kolors_quality_benchmark.py --compiler oneflow --dataset coco --csv-file resources/MS-COCO_val2014_30k_captions.csv --output-dir /path/to/your/kolors_oneflow_coco
python3 models/kolors/text_to_image_kolors_quality_benchmark.py \
--compiler oneflow \
--dataset coco \
--csv-file resources/MS-COCO_val2014_30k_captions.csv \
--output-dir /path/to/your/kolors_oneflow_coco
```
```
# Accelerate using onediff's nexfort backend.
python3 models/kolors/text_to_image_kolors_quality_benchmark.py --compiler nexfort --compiler-config '{"mode": "max-optimize:max-autotune:low-precision", "memory_format": "channels_last"}' --dataset coco --csv-file resources/MS-COCO_val2014_30k_captions.csv --output-dir /path/to/your/kolors_nexfort_coco
python3 models/kolors/text_to_image_kolors_quality_benchmark.py \
--compiler nexfort \
--compiler-config '{"mode": "max-optimize:max-autotune:low-precision", "memory_format": "channels_last"}' \
--dataset coco \
--csv-file resources/MS-COCO_val2014_30k_captions.csv \
--output-dir /path/to/your/kolors_nexfort_coco
```
- On Human Preference Dataset v2 (HPD v2):
Simply modify the `--dataset` parameters, do not read prompts from the `--csv-file` parameter, and customize the `--output-dir` for generating images. For example:
```
python3 models/kolors/text_to_image_kolors_quality_benchmark.py --dataset hps --output-dir /path/to/your/kolors_torch_hps
python3 models/kolors/text_to_image_kolors_quality_benchmark.py \
--dataset hps \
--output-dir /path/to/your/kolors_torch_hps
```
### 2. Test using multiple indicators with scripts.
Expand Down

0 comments on commit 2808806

Please sign in to comment.