Hidream I1 Full FP8

🚀 HiDream-I1 - 開源圖像生成基礎模型

HiDream-I1 是一款全新的開源圖像生成基礎模型，擁有 170 億參數，能在數秒內實現頂尖的圖像生成質量。如需瞭解更多特性並體驗我們產品的完整功能，請訪問 https://vivago.ai/。

✨ 主要特性

• ✨ 卓越的圖像質量 - 在多種風格（包括逼真照片、卡通、藝術風格等）下都能產生出色的結果。達到了最先進的 HPS v2.1 評分，符合人類偏好。
• 🎯 一流的提示遵循能力 - 在 GenEval 和 DPG 基準測試中取得了行業領先的分數，超越了所有其他開源模型。
• 🔓 開源 - 在 MIT 許可下發布，以促進科學進步並推動創新。
• 💼 商業友好 - 生成的圖像可自由用於個人項目、科學研究和商業應用。

🚀 快速開始

請確保你已經安裝了 Flash Attention。手動安裝時，我們建議使用 CUDA 版本 12.4。

pip install -r requirements.txt

克隆 GitHub 倉庫：

git clone https://github.com/HiDream-ai/HiDream-I1

然後你可以運行推理腳本來生成圖像：

# 全量模型推理
python ./inference.py --model_type full

# 蒸餾開發模型推理
python ./inference.py --model_type dev

# 蒸餾快速模型推理
python ./inference.py --model_type fast

⚠️ 重要提示

推理腳本將自動下載 meta-llama/Meta-Llama-3.1-8B-Instruct 模型文件。如果你遇到網絡問題，可以提前下載這些文件並將它們放在適當的緩存目錄中，以避免推理期間下載失敗。

💻 使用示例

基礎用法

# 全量模型推理
python ./inference.py --model_type full

# 蒸餾開發模型推理
python ./inference.py --model_type dev

# 蒸餾快速模型推理
python ./inference.py --model_type fast

高級用法

# 運行 Gradio 演示進行交互式圖像生成
python gradio_demo.py 

📚 詳細文檔

評估指標

DPG-Bench

GenEval

HPSv2.1 基準測試

📄 許可證

本倉庫中的 Transformer 模型遵循 MIT 許可證。VAE 來自 FLUX.1 [schnell]，文本編碼器來自 google/t5-v1_1-xxl 和 meta-llama/Meta-Llama-3.1-8B-Instruct。請遵循這些組件指定的許可條款。你擁有使用此模型創建的所有內容。你可以自由使用生成的內容，但必須遵守本許可協議。你應對模型的使用負責。請勿創建非法內容、有害材料、可能傷害他人的個人信息、虛假信息或針對弱勢群體的內容。

致謝

• VAE 組件來自 FLUX.1 [schnell]，遵循 Apache 2.0 許可證。
• 文本編碼器來自 google/t5-v1_1-xxl（遵循 Apache 2.0 許可證）和 meta-llama/Meta-Llama-3.1-8B-Instruct（遵循 Llama 3.1 社區許可協議）。