Instructions to use ziheng1234/ImageCritic with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use ziheng1234/ImageCritic with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline from diffusers.utils import load_image # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("ziheng1234/ImageCritic", dtype=torch.bfloat16, device_map="cuda") prompt = "Turn this cat into a dog" input_image = load_image("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/cat.png") image = pipe(image=input_image, prompt=prompt).images[0] - Notebooks
- Google Colab
- Kaggle
Update README.md
Browse files
README.md
CHANGED
|
@@ -16,86 +16,19 @@ The model was presented in the paper [The Consistency Critic: Correcting Inconsi
|
|
| 16 |
* 🤗 [Hugging Face Space Demo](https://huggingface.co/spaces/ziheng1234/ImageCritic)
|
| 17 |
* 📦 [Hugging Face Dataset](https://huggingface.co/datasets/ziheng1234/Critic-10K)
|
| 18 |
|
| 19 |
-
<img src='https://
|
| 20 |
|
| 21 |
## 🖼️ Visual Results
|
| 22 |
|
| 23 |
ImageCritic can effectively resolve detail-related issues in various customized generation scenarios, providing significant improvements over existing methods.
|
| 24 |
|
| 25 |
-
<img src='https://github.com/HVision-NKU/ImageCritic/
|
| 26 |
-
|
| 27 |
-
## 🔧 Dependencies and Installation
|
| 28 |
-
|
| 29 |
-
We recommend using Python 3.10 and PyTorch with CUDA support. To set up the environment:
|
| 30 |
-
|
| 31 |
-
```bash
|
| 32 |
-
# Create a new conda environment
|
| 33 |
-
conda create -n imagecritic python=3.10
|
| 34 |
-
conda activate imagecritic
|
| 35 |
-
|
| 36 |
-
# Install other dependencies
|
| 37 |
-
pip install -r requirements.txt
|
| 38 |
-
```
|
| 39 |
-
|
| 40 |
-
## ⚡ Quick Inference
|
| 41 |
-
### Tips
|
| 42 |
-
Due to copyright issues, we have embedded the download of the kontext model weights in the inference code below, You can run following inference code directly.
|
| 43 |
-
If you have already downloaded the corresponding model, you can comment out the related code and directly replace the inference path.
|
| 44 |
-
|
| 45 |
-
### Single case inference
|
| 46 |
-
```bash
|
| 47 |
-
python infer.py
|
| 48 |
-
```
|
| 49 |
-
|
| 50 |
-
### Local Gradio Demo
|
| 51 |
-
```bash
|
| 52 |
-
python app.py
|
| 53 |
-
```
|
| 54 |
-
|
| 55 |
-
### Single Model Download
|
| 56 |
-
You can download the base model FLUX.1-Kontext-dev directly from [Hugging Face](https://huggingface.co/black-forest-labs/FLUX.1-Kontext-dev).
|
| 57 |
-
|
| 58 |
-
Alternatively, you can download it via the following command
|
| 59 |
-
(⚠️ Remember to replace `your_hf_token` in the script with your actual Hugging Face access token):
|
| 60 |
-
```bash
|
| 61 |
-
python ./download_kontext.py
|
| 62 |
-
```
|
| 63 |
-
|
| 64 |
-
You can download our ImageCritic directly from [Hugging Face](https://huggingface.co/ziheng1234/ImageCritic).
|
| 65 |
-
|
| 66 |
-
Alternatively, you can download it via following code:
|
| 67 |
-
|
| 68 |
-
```bash
|
| 69 |
-
python ./download_imageCritic.py
|
| 70 |
-
```
|
| 71 |
-
Or using Git:
|
| 72 |
-
```bash
|
| 73 |
-
git lfs install
|
| 74 |
-
git clone https://huggingface.co/ziheng1234/ImageCritic
|
| 75 |
-
```
|
| 76 |
-
|
| 77 |
-
## Dataset Download
|
| 78 |
-
|
| 79 |
-
You can download our training dataset Critic-10K directly from [Hugging Face](https://huggingface.co/datasets/ziheng1234/Critic-10K).
|
| 80 |
-
|
| 81 |
-
Alternatively, you can download it via Python:
|
| 82 |
-
|
| 83 |
-
```bash
|
| 84 |
-
python /raid/users/oyzh/ImageCritic/download_dataset.py
|
| 85 |
-
```
|
| 86 |
-
Or using Git:
|
| 87 |
-
```bash
|
| 88 |
-
git lfs install
|
| 89 |
-
git clone https://huggingface.co/datasets/ziheng1234/Critic-10K
|
| 90 |
-
```
|
| 91 |
|
| 92 |
### Online HuggingFace Demo
|
| 93 |
You can try ImageCritic demo on [HuggingFace](https://huggingface.co/spaces/ziheng1234/ImageCritic).
|
| 94 |
|
| 95 |
## Citation
|
| 96 |
|
| 97 |
-
If ImageCritic is helpful, please help to ⭐ the repo.
|
| 98 |
-
|
| 99 |
If you find this project useful for your research, please consider citing our paper:
|
| 100 |
```bibtex
|
| 101 |
@article{ouyang2025consistency,
|
|
|
|
| 16 |
* 🤗 [Hugging Face Space Demo](https://huggingface.co/spaces/ziheng1234/ImageCritic)
|
| 17 |
* 📦 [Hugging Face Dataset](https://huggingface.co/datasets/ziheng1234/Critic-10K)
|
| 18 |
|
| 19 |
+
<img src='https://raw.githubusercontent.com/HVision-NKU/ImageCritic/refs/heads/master/figure/teaser.png' width='100%' />
|
| 20 |
|
| 21 |
## 🖼️ Visual Results
|
| 22 |
|
| 23 |
ImageCritic can effectively resolve detail-related issues in various customized generation scenarios, providing significant improvements over existing methods.
|
| 24 |
|
| 25 |
+
<img src='https://github.com/HVision-NKU/ImageCritic/blob/master/figure/compare.png?raw=true' width='100%' />
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 26 |
|
| 27 |
### Online HuggingFace Demo
|
| 28 |
You can try ImageCritic demo on [HuggingFace](https://huggingface.co/spaces/ziheng1234/ImageCritic).
|
| 29 |
|
| 30 |
## Citation
|
| 31 |
|
|
|
|
|
|
|
| 32 |
If you find this project useful for your research, please consider citing our paper:
|
| 33 |
```bibtex
|
| 34 |
@article{ouyang2025consistency,
|