Add `library_name` metadata tag

This PR enhances the model card by adding the `library_name: diffusers` metadata tag. This will enable the automatic "How to use" widget on the Hugging Face Hub, providing users with a ready-to-run code snippet for the model, as it is compatible with the `diffusers` library.

Files changed (1) hide show

README.md +13 -204

README.md CHANGED Viewed

@@ -1,37 +1,22 @@
 ---
-license: apache-2.0
 language:
 - en
 pipeline_tag: image-to-image
 ---
 # End2End Virtual Tryon with Visual Reference
 [![Hugging Face](https://img.shields.io/badge/Hugging%20Face-EVTAR-ff9900?style=flat)](https://huggingface.co/qihoo360/EVTAR) [![arXiv](https://img.shields.io/badge/arXiv-2101.00001-B31B1B?style=flat)](https://arxiv.org/abs/2511.00956)
 ![examples](examples.png)
 We propose **EVTAR**, an End-to-End Virtual Try-on model with Additional Visual Reference, that directly fits the target garment onto the person image while incorporating reference images to enhance the model's ability to preserve and accurately depict clothing details.
 ## 💡 Github
 [EVTAR](https://github.com/360CVGroup/EVTAR)
 ## 💡 Pretrained Models
 We provide pretrained backbone networks and LoRA weights for testing and deployment. Please download the `.safetensors` files from [here] and place them in the `checkpoints` directory.
@@ -41,126 +26,43 @@ We provide pretrained backbone networks and LoRA weights for testing and deploym
 1024_768_pytorch_lora_weights.safetensors:1024x768 resolution high-quality virtual fitting model
 ✅ **Available**
 ## 💡 Update
 - [x] [2025.10.11] Release the virtual try-on inference code and LoRA weights.
 - [x] [2025.10.13] Release the technical report on Arxiv.
 ## 💪 Highlight Feature
--  **And End-To-End virtual try-on model:** Can function either as an inpainting model for placing the target clothing into masked areas, or as a direct garment transfer onto the human body.
--  **Using Reference Image To Enhance the Try-on Performance:** To emulate human attention on the overall wearing effect rather than the garment itself when shopping online, our model allows using images of a model wearing the target clothing as input, thereby better preserving its material texture and design details.
--  **Improved Performance** Our model achieves state-of-the-art performance on public benchmarks and demonstrates strong generalization ability to in-the-wild inputs.
 ## 🧩 Environment Setup
 ```
 conda create -n EVTAR python=3.12 -y
 conda activate EVTAR
 pip install -r requirements.txt
 ```
 ## 📂 Preparation of Dataset and Pretrained Models
 ### Dataset
 Currently, we provide a small test set with additional reference images "difference person wearing the target cloth" for trying our model. We plan to release the reference data generation code, along with our proposed full dataset containing model reference images, in the future.
 Nevertheless, inference can still be performed in a reference-free setting on public benchmarks, including [VITON-HD](https://github.com/shadow2496/VITON-HD) and [DressCode](https://github.com/aimagelab/dress-code).
 ### Reference Data Preparation
 One key feature of our method is the use of _reference data_, where an image of a different person wearing the target garment is provided to help the model imagine how the target person would look in that garment. In most online shopping applications, such additonal reference images are commonly used by customers to better visualize the clothing. However, publicly available datasets such as VITON-HD and DressCode do not include such reference data, so we generate them ourselves.
 Please prepare the pretrained weights of the Flux-Kontext model and the Qwen2.5-VL-32B model. And you can generate the additonal reference image using the following commands:
 ```
 accelerate launch --num_processes 8 --main_process_port 29500 generate_reference.py \
 --instance_data_dir "path_to_your_datasets" \
@@ -169,40 +71,14 @@ accelerate launch --num_processes 8 --main_process_port 29500 generate_reference
 --desc_path "desc.json"
 ```
 ### Pretrained Models
 We provide pretrained backbone networks and LoRA weights for testing and deployment. Please download the `.safetensors` files from [here] and place them in the `checkpoints` directory.
 ## ⏳ Inference Pipeline
 Here we provide the inference code for our EVTAR.
 ```
 accelerate launch --num_processes 8 --main_process_port 29500 inference.py \
@@ -221,62 +97,24 @@ accelerate launch --num_processes 8 --main_process_port 29500 inference.py \
 --use_person
 ```
--  `pretrained_model_name_or_path`: Path to the downloaded Flux-Kontext model weights.
--  `instance_data_dir`: Path to your dataset. For inference on VITON-HD or DressCode, ensure that the words "viton" or "DressCode" appear in the path.
--  `output_dir`: Path to the downloaded or trained LoRA weights.
--  `cond_scale`: Resize scale of the reference image during training. Defaults to `1.0` for $512\times384$ and `2.0` for $1024\times768$ resolution.
--  `use_reference`: Whether to use a additonal reference image as input.
--  `use_different`: **Only applicable for VITON/DressCode inference.** Whether to use different cloth-person pairs.
--  `use_person`: **Only applicable for VITON/DressCode inference.** Whether to use the unmasked person image instead of the agnostic masked image as input for the virtual try-on task.
 ## 📊 Evaluation
 We quantitatively evaluate the quality of virtual try-on results using the FID, KID, SSIM, and LPIPS. Here, we provide the evaluation code for the VITON-HD and DressCode datasets.
 ```
 # Evaluation on VITON-HD dataset
@@ -287,12 +125,6 @@ CUDA_VISIBLE_DEVICES=0 python eval_dresscode.py \
 --paired
 ```
 ```
 # Evaluation on DressCode dataset
@@ -302,45 +134,22 @@ CUDA_VISIBLE_DEVICES=0 python eval.py \
 --pred_folder_base [[path_to_your_generated_image_folder]]\
 ```
--  `paired`: If you perform unpaired generation, where different garments are fitted onto the target person, you should enable this flag during evaluation.
 Evaluation result on VITON-HD dataset:
 ![examples](VITON_results.png)
 Evaluation result on DressCode dataset:
 ![examples](DressCode_results.png)
 ## 🌸 Acknowledgement
 This code is mainly built upon [Diffusers](https://github.com/huggingface/diffusers/tree/main), [Flux](https://github.com/huggingface/diffusers/tree/main/src/diffusers/pipelines/flux), and [CatVTON](https://github.com/Zheng-Chong/CatVTON/) repositories. Thanks so much for their solid work!
 ## 💖 Citation
 If you find this repository useful, please consider citing our paper:
 ```
 @misc{li2025evtarendtoendtryadditional,

 ---
 language:
 - en
+license: apache-2.0
 pipeline_tag: image-to-image
+library_name: diffusers
 ---
 # End2End Virtual Tryon with Visual Reference
 [![Hugging Face](https://img.shields.io/badge/Hugging%20Face-EVTAR-ff9900?style=flat)](https://huggingface.co/qihoo360/EVTAR) [![arXiv](https://img.shields.io/badge/arXiv-2101.00001-B31B1B?style=flat)](https://arxiv.org/abs/2511.00956)
 ![examples](examples.png)
 We propose **EVTAR**, an End-to-End Virtual Try-on model with Additional Visual Reference, that directly fits the target garment onto the person image while incorporating reference images to enhance the model's ability to preserve and accurately depict clothing details.
 ## 💡 Github
 [EVTAR](https://github.com/360CVGroup/EVTAR)
 ## 💡 Pretrained Models
 We provide pretrained backbone networks and LoRA weights for testing and deployment. Please download the `.safetensors` files from [here] and place them in the `checkpoints` directory.
 1024_768_pytorch_lora_weights.safetensors:1024x768 resolution high-quality virtual fitting model
 ✅ **Available**
 ## 💡 Update
 - [x] [2025.10.11] Release the virtual try-on inference code and LoRA weights.
 - [x] [2025.10.13] Release the technical report on Arxiv.
 ## 💪 Highlight Feature
+- **And End-To-End virtual try-on model:** Can function either as an inpainting model for placing the target clothing into masked areas, or as a direct garment transfer onto the human body.
+- **Using Reference Image To Enhance the Try-on Performance:** To emulate human attention on the overall wearing effect rather than the garment itself when shopping online, our model allows using images of a model wearing the target clothing as input, thereby better preserving its material texture and design details.
+- **Improved Performance** Our model achieves state-of-the-art performance on public benchmarks and demonstrates strong generalization ability to in-the-wild inputs.
 ## 🧩 Environment Setup
 ```
 conda create -n EVTAR python=3.12 -y
 conda activate EVTAR
 pip install -r requirements.txt
 ```
 ## 📂 Preparation of Dataset and Pretrained Models
 ### Dataset
 Currently, we provide a small test set with additional reference images "difference person wearing the target cloth" for trying our model. We plan to release the reference data generation code, along with our proposed full dataset containing model reference images, in the future.
 Nevertheless, inference can still be performed in a reference-free setting on public benchmarks, including [VITON-HD](https://github.com/shadow2496/VITON-HD) and [DressCode](https://github.com/aimagelab/dress-code).
 ### Reference Data Preparation
 One key feature of our method is the use of _reference data_, where an image of a different person wearing the target garment is provided to help the model imagine how the target person would look in that garment. In most online shopping applications, such additonal reference images are commonly used by customers to better visualize the clothing. However, publicly available datasets such as VITON-HD and DressCode do not include such reference data, so we generate them ourselves.
 Please prepare the pretrained weights of the Flux-Kontext model and the Qwen2.5-VL-32B model. And you can generate the additonal reference image using the following commands:
 ```
 accelerate launch --num_processes 8 --main_process_port 29500 generate_reference.py \
 --instance_data_dir "path_to_your_datasets" \
 --desc_path "desc.json"
 ```
 ### Pretrained Models
 We provide pretrained backbone networks and LoRA weights for testing and deployment. Please download the `.safetensors` files from [here] and place them in the `checkpoints` directory.
 ## ⏳ Inference Pipeline
 Here we provide the inference code for our EVTAR.
 ```
 accelerate launch --num_processes 8 --main_process_port 29500 inference.py \
 --use_person
 ```
+- `pretrained_model_name_or_path`: Path to the downloaded Flux-Kontext model weights.
+- `instance_data_dir`: Path to your dataset. For inference on VITON-HD or DressCode, ensure that the words "viton" or "DressCode" appear in the path.
+- `output_dir`: Path to the downloaded or trained LoRA weights.
+- `cond_scale`: Resize scale of the reference image during training. Defaults to `1.0` for $512\times384$ and `2.0` for $1024\times768$ resolution.
+- `use_reference`: Whether to use a additonal reference image as input.
+- `use_different`: **Only applicable for VITON/DressCode inference.** Whether to use different cloth-person pairs.
+- `use_person`: **Only applicable for VITON/DressCode inference.** Whether to use the unmasked person image instead of the agnostic masked image as input for the virtual try-on task.
 ## 📊 Evaluation
 We quantitatively evaluate the quality of virtual try-on results using the FID, KID, SSIM, and LPIPS. Here, we provide the evaluation code for the VITON-HD and DressCode datasets.
 ```
 # Evaluation on VITON-HD dataset
 --paired
 ```
 ```
 # Evaluation on DressCode dataset
 --pred_folder_base [[path_to_your_generated_image_folder]]\
 ```
+- `paired`: If you perform unpaired generation, where different garments are fitted onto the target person, you should enable this flag during evaluation.
 Evaluation result on VITON-HD dataset:
 ![examples](VITON_results.png)
 Evaluation result on DressCode dataset:
 ![examples](DressCode_results.png)
 ## 🌸 Acknowledgement
 This code is mainly built upon [Diffusers](https://github.com/huggingface/diffusers/tree/main), [Flux](https://github.com/huggingface/diffusers/tree/main/src/diffusers/pipelines/flux), and [CatVTON](https://github.com/Zheng-Chong/CatVTON/) repositories. Thanks so much for their solid work!
 ## 💖 Citation
 If you find this repository useful, please consider citing our paper:
 ```
 @misc{li2025evtarendtoendtryadditional,