Zenn
👩‍💻

UbuntuのローカルPCでComfyUIにFlux.1を導入する方法

2024/10/23に公開
Jupyter Notebook
Ubuntu
flux
JupyterLab
ComfyUI
tech

はじめに

この記事では、ローカルPCのUbuntu環境でSSH接続を使用し、Jupyter Labを通じてComfyUIにFlux.1-devモデルを導入する方法を詳しく解説します。Flux.1-devは高品質な画像生成が可能な最新のAIモデルで、ComfyUIと組み合わせることで強力な画像生成ツールとなります。

前提条件

  • UbuntuがインストールされたローカルPC
  • SSH接続の設定
  • Jupyter Labのインストールと設定
  • ComfyUIのインストール
  • Hugging Faceアカウントとアクセストークン

ComfyUIのGithubリポジトリにあるJupyter Notebookは、paperspaceやGoogle colab用なので少し手を加える必要があります。

手順

1. 必要なライブラリのインストール

まず、必要なライブラリをインストールします。Jupyter Labのセルで以下のコマンドを実行してください：

!pip install huggingface_hub

2. Hugging Faceへのログイン

Hugging Faceにログインするために、以下のコードを実行します：

from huggingface_hub import login
login(token="hf_あなたのトークン")

注意："hf_あなたのトークン"の部分は、実際のHugging Faceアクセストークンに置き換えてください。

FLUX.1-devは事前にアクセス申請が必要です。
申請後、トークンを作成してRepositories permissionsにFLUX.1-devを追加します。

3. ダウンロード用の関数とパスの設定

以下のコードで、ファイルのダウンロード用の関数とComfyUIのディレクトリパスを設定します：

import os
from huggingface_hub import hf_hub_download

# ホームディレクトリのパスを取得
home_dir = os.path.expanduser("~")
comfyui_dir = os.path.join(home_dir, "ComfyUI")

4. Flux1.1モデルのダウンロード

Flux1.1モデルをダウンロードするために、以下のコードを実行します：

# Flux1.1のモデルリスト
flux_models = [
    ("black-forest-labs/FLUX.1-dev", "flux1-dev.safetensors", "models/unet"),
    ("comfyanonymous/flux_text_encoders", "t5xxl_fp16.safetensors", "models/clip"),
    ("comfyanonymous/flux_text_encoders", "clip_l.safetensors", "models/clip"),
    ("black-forest-labs/FLUX.1-schnell", "ae.safetensors", "models/vae")
]

# モデルのダウンロード
print("Flux1.1のモデルのダウンロードを始めます")
for repo_id, filename, local_dir in flux_models:
    local_path = os.path.join(comfyui_dir, local_dir, filename)
    hf_hub_download(repo_id=repo_id, filename=filename, local_dir=os.path.join(comfyui_dir, local_dir))
    print(f"ダウンロード完了: {local_path}")

print("全てのモデルのダウンロードが完了しました")

5. ComfyUIの設定

ダウンロードしたモデルファイルをComfyUIで使用するために、ComfyUIの設定を更新する必要があります。具体的な手順はComfyUIのドキュメントを参照してください。

まとめ

以上の手順で、UbuntuのローカルPC環境でComfyUIにFlux1.1モデルを導入することができます。この方法を使えば、Google ColabやPaperspaceなどのクラウドサービスに頼ることなく、自分のPCでFlux1.1を使った高品質な画像生成が可能になります。

ただし、Flux1.1モデルは大容量のファイルをダウンロードするため、十分なストレージ容量と高速なインターネット接続が必要です。また、モデルの使用にはある程度のGPU性能も要求されますので、ハードウェア要件も確認してください。

https://huggingface.co/black-forest-labs/FLUX.1-dev/tree/main

https://huggingface.co/comfyanonymous/flux_text_encoders

https://huggingface.co/black-forest-labs/FLUX.1-schnell/tree/main

Discussion