Skip to content

beaugogh/diffusions

Repository files navigation

diffusions

Installation

system prerequisites:

  1. install CUDA toolkit, see here
  2. if not already done, add cuda bin to path by appending export PATH=/usr/local/cuda/bin${PATH:+:${PATH}} to the end of ~/.bashrc, then do source ~/.bashrc to refresh. Finally verify with nvcc -V.
  3. install ffmpeg
sudo apt install ffmpeg
  1. install rustup compiler
curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh

python environment:

cd path/to/diffusions
python -m venv venv
source venv/bin/activate

pip install -r requirements.txt

Note: LivePortrait requirements include transformers, exclude that!

Model Weights

download LivePortrait model (see here)

Note that the downloaded models need to be put into ./LivePortrait/pretrained_weights, the default model directory can be modified in ./LivePortrait/src/config/inference_config.py, but not necessary. For some reason, the huggingface repo does not contain all the model files, you can just put the downloaed liveportrait models, the script will automatically download the rest, more complete version of models.

huggingface-cli download KwaiVGI/LivePortrait --local-dir ./models/live_portrait_hf --exclude "*.git*" "README.md" "docs"

download DensePose model (see here)

wget https://dl.fbaipublicfiles.com/densepose/densepose_rcnn_R_50_FPN_s1x/165712039/model_final_162be9.pkl

Quickfixes

Nvidia issue after suspension

sudo apt install nvidia-modprobe

sudo rmmod nvidia_uvm && sudo modprobe nvidia_uvm

TODOs

Repos

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published