update readme

README.md

@@ -58,9 +58,16 @@ This repo contains PyTorch model definitions, pretrained weights and inference/s
   - [Contents](#contents)
   - [Abstract](#abstract)
   - [HunyuanImage-2.1 Overall Pipeline](#hunyuanimage-21-overall-pipeline)
+    - [Training Data and Caption](#training-data-and-caption)
+    - [Text-to-Image Model Architecture](#text-to-image-model-architecture)
+    - [Reinforcement Learning from Human Feedback](#reinforcement-learning-from-human-feedback)
+    - [Rewriting Model](#rewriting-model)
+    - [Model distillation](#model-distillation)
   - [🎉 HunyuanImage-2.1 Key Features](#-hunyuanimage-21-key-features)
   - [Prompt Enhanced Demo](#prompt-enhanced-demo)
   - [📈 Comparisons](#-comparisons)
+    - [SSAE Evaluation](#ssae-evaluation)
+    - [GSB Evaluation](#gsb-evaluation)
   - [📜 System Requirements](#-system-requirements)
   - [🛠️ Dependencies and Installation](#️-dependencies-and-installation)
   - [🧱 Download Pretrained Models](#-download-pretrained-models)
@@ -231,7 +238,7 @@ image = pipe(
     width=2048,
     height=2048,
     use_reprompt=True,  # Enable prompt enhancement
-    use_refiner=False,  # Refiner is not supported yet; coming soon.
+    use_refiner=False,  # Disable refiner model, enable it to achieve higher image quality.
     # For the distilled model, use 8 steps for faster inference.
     # For the non-distilled model, use 50 steps for better quality
     num_inference_steps=8 if "distilled" in model_name else 50, 

README_CN.md

@@ -32,21 +32,29 @@
 
 
 ## 目录
-- [🔥🔥🔥 最新动态](#-最新动态)
-- [🎥 示例](#-示例)
-- [目录](#目录)
-- [摘要](#摘要)
-- [混元图像 2.1 整体流程](#混元图像-21-整体流程)
-- [🎉 关键特性](#-关键特性)
-- [提示词增强演示](#提示词增强示例)
-- [📈 对比](#-对比)
-- [📜 系统要求](#-系统要求)
-- [🛠️ 依赖与安装](#-依赖与安装)
-- [🧱 模型下载](#-模型下载)
-- [🔑 使用](#-使用)
-- [🔗 BibTeX](#-bibtex)
-- [致谢](#致谢)
-- [Github Star 历史](#github-star-历史)
+- [混元图像 2.1：一种用于高分辨率（2K）文本到图像生成的高效扩散模型](#混元图像-21一种用于高分辨率2k文本到图像生成的高效扩散模型)
+  - [🔥🔥🔥 最新动态](#-最新动态)
+  - [🎥 示例](#-示例)
+  - [目录](#目录)
+  - [摘要](#摘要)
+  - [混元图像 2.1 整体流程](#混元图像-21-整体流程)
+    - [训练数据和标注](#训练数据和标注)
+    - [文本到图像模型架构](#文本到图像模型架构)
+    - [人类反馈强化学习](#人类反馈强化学习)
+    - [改写模型](#改写模型)
+    - [模型蒸馏](#模型蒸馏)
+  - [🎉 关键特性](#-关键特性)
+  - [提示词增强示例](#提示词增强示例)
+  - [📈 对比](#-对比)
+    - [SSAE 评测](#ssae-评测)
+    - [GSB 评测](#gsb-评测)
+  - [📜 系统要求](#-系统要求)
+  - [🛠️ 依赖与安装](#️-依赖与安装)
+  - [🧱 模型下载](#-模型下载)
+  - [🔑 使用](#-使用)
+  - [🔗 BibTeX](#-bibtex)
+  - [致谢](#致谢)
+  - [Github Star 历史](#github-star-历史)
 
 ---
 
@@ -203,7 +211,7 @@ image = pipe(
     width=2048,
     height=2048,
     use_reprompt=True,  # 启用提示词增强
-    use_refiner=False,  # 精修（Refiner）功能暂不支持，后续上线
+    use_refiner=False,  # 关闭精修模型, 可开启以获得更高画质
     # 对于蒸馏版模型，建议使用 8 步以加快推理速度；
     # 对于非蒸馏版模型，建议使用 50 步以获得更高画质
     num_inference_steps=8 if "distilled" in model_name else 50,