diff --git a/docker-compose.yml b/docker-compose.yml index 699cfc6bb66..a4b7648788b 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -12,6 +12,7 @@ services: - ./input:/app/input - ./output:/app/output - ../models:/app/models + - ../models/u2net:/root/.u2net - ./custom_nodes:/app/custom_nodes tty: true deploy: diff --git a/requirements_docker.txt b/requirements_docker.txt index 5fee0437338..69cf837f487 100644 --- a/requirements_docker.txt +++ b/requirements_docker.txt @@ -4,7 +4,7 @@ PyGithub matrix-client==0.4.0 transformers huggingface-hub>0.20 -typer +typer==0.9.0 rich typing-extensions @@ -50,8 +50,26 @@ cachetools yacs mediapipe svglib - +fvcore +omegaconf imageio-ffmpeg librosa color-matcher -huggingface_hub \ No newline at end of file +huggingface_hub +filelock +ftfy +trimesh[easy] +albumentations +scikit-learn + +# AnimateDiff +pandas +numexpr + +#MimicMotioin +diffusers +decord +einops +moviepy +accelerate +av \ No newline at end of file