🎏

StreamDiffusionでVideo2Video（ローカル）

2024/01/24に公開

StreamDiffusionをローカルで動かす

Google Colabに続いてローカルでStreamDiffusionを動かす方法です。

上記設定の上でセットアップしていきます。

$ cd && git clone https://github.com/cumulo-autumn/StreamDiffusion.git
$ cd StreamDiffusion
$ docker build -t stream-diffusion:latest -f Dockerfile .

StreamDiffusion/demo/vid2vid/app.pyを以下の通り修正します。

- demo.launch()
+ demo.launch(server_name="0.0.0.0")

必要に応じてpromptも好きに修正します。

ここまでは一回実行すればOKです。

Dockerを起動します。

$ ~/StreamDiffusion
$ docker run -p 7860:7860 --gpus all -it -v $(pwd):/home/ubuntu/streamdiffusion stream-diffusion:latest

コンテナ内で以下実行します。

# cd demo/vid2vid/
# pip3 install -r requirements.txt
# python3 app.py

以下URLにブラウザでアクセス

http://127.0.0.1:7860/

Gradioのデモアプリが起動するので、動画アップロードして変換などしましょう。