Hidream I1 Fast

🚀 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

高度な使用法

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