Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
teowu committed Jan 31, 2024
1 parent 4007d89 commit b938a83
Showing 1 changed file with 12 additions and 1 deletion.
13 changes: 12 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -59,7 +59,7 @@
</div>
</div>

## [Important Note!] v1.1 <-> transformers==4.36.1
## [Important Note!] v1.1 <-> transformers==4.36.1+

This LLaVA-style repository has been built on `transformers==4.31.0`, which is incompatible with many new models available on Hugging Face. This requires to build a separate environment for the MLLM/LMM repository, which is somewhat troublesome for this **visual scoring** model, as we expect the **Q-Align/OneAlign** to effectively boost other disciplines (image/video generation, *etc*). Both the repository and the **AutoModel** (as follows) are updated to the newest version.

Expand Down Expand Up @@ -87,6 +87,17 @@ model.score([Image.open(requests.get("https://raw.githubusercontent.com/Q-Future
stream=True).raw)], task_="quality", input_="image") # task_ : quality | aesthetics; # input_: image | video
```

This model also supports to be instantialized via latest [IQA-Pytorch](https://github.com/chaofengc/IQA-PyTorch):

```python
import pyiqa
import torch

qalign = pyiqa.create_metric('qalign').cuda()
quality_score = qalign(input, task_='quality')
aesthetic_score = qalign(input, task_='aesthetic')
```

## Installation

If you only need to infer (or evaluate):
Expand Down

0 comments on commit b938a83

Please sign in to comment.