Hidream I1 Dev

🚀 HiDream-I1

HiDream-I1は、170億のパラメータを持つ新しいオープンソースの画像生成基礎モデルで、数秒で最先端の画像生成品質を実現します。

より多くの機能を体験し、当社製品の全機能を試すには、https://vivago.ai/をご覧ください。

🚀 クイックスタート

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モデルファイルをダウンロードします。ネットワークの問題が発生した場合は、事前にこれらのファイルをダウンロードし、適切なキャッシュディレクトリに配置することで、推論中のダウンロード失敗を回避できます。

✨ 主な機能

• ✨ 卓越した画像品質 - 写実的、漫画、アートなど、複数のスタイルで卓越した結果を生み出します。最先端のHPS v2.1スコアを達成し、人間の好みに合致します。
• 🎯 最高水準のプロンプト追従性 - GenEvalとDPGベンチマークで業界最高のスコアを達成し、他のすべてのオープンソースモデルを上回ります。
• 🔓 オープンソース - 科学の進歩を促進し、創造的な革新を可能にするために、MITライセンスの下で公開されています。
• 💼 商用利用にも適しています - 生成された画像は、個人プロジェクト、科学研究、商用アプリケーションに自由に使用できます。

📦 インストール

依存関係のインストール

pip install -r requirements.txt

リポジトリのクローン

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

📚 ドキュメント

プロジェクトの更新情報

• 🚀 2025年4月28日：命令ベースの画像編集モデルHiDream-E1-Fullをオープンソース化しました。https://huggingface.co/spaces/HiDream-ai/HiDream-E1-Fullで体験してください！

評価指標

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コミュニティライセンス契約）から取得されています。