Skip to content

Commit

Permalink
chore(runner): remove duplicate pynvml dep
Browse files Browse the repository at this point in the history
This commit removes the duplicate pynvml dependency.
  • Loading branch information
rickstaa committed Dec 9, 2024
1 parent a579cf3 commit 2ff0f77
Show file tree
Hide file tree
Showing 5 changed files with 6 additions and 8 deletions.
4 changes: 2 additions & 2 deletions runner/docker/Dockerfile.live-base-comfyui
Original file line number Diff line number Diff line change
Expand Up @@ -35,14 +35,14 @@ RUN cd /comfyui/custom_nodes && \
cd ComfyUI-Misc-Effects && \
git checkout c6b360c78611134c3723388170475eb4898ff6b7

RUN pip install torch==2.5.1 torchvision torchaudio tqdm nvidia-ml-py==12.560.30
RUN pip install torch==2.5.1 torchvision torchaudio tqdm

# Install comfystream (which includes ComfyUI)
RUN pip install git+https://github.com/yondonfu/comfystream.git
RUN git clone https://github.com/yondonfu/comfystream.git && \
cd comfystream && \
pip install -r requirements.txt && \
cp -r nodes/tensor_utils /comfyui/custom_nodes/
cp -r nodes/tensor_utils /comfyui/custom_nodes/

# Install ComfyUI-SAM2-Realtime (https://github.com/pschroedl/ComfyUI-SAM2-Realtime.git)
RUN cd /comfyui/custom_nodes && \
Expand Down
3 changes: 1 addition & 2 deletions runner/docker/Dockerfile.live-base-streamdiffusion
Original file line number Diff line number Diff line change
Expand Up @@ -21,8 +21,7 @@ RUN pip install --no-cache-dir \
RUN pip install --no-cache-dir \
huggingface-hub==0.23.2 \
diffusers==0.30.0 \
protobuf==5.27.2 \
nvidia-ml-py==12.560.30
protobuf==5.27.2

# Install StreamDiffusion @ 765d71029b1404b94aee2865178d71c257c20318 (latest at time of writing)
RUN pip install git+https://github.com/cumulo-autumn/StreamDiffusion.git@765d710#egg=streamdiffusion[tensorrt]
Expand Down
2 changes: 1 addition & 1 deletion runner/requirements-liveportrait.txt
Original file line number Diff line number Diff line change
Expand Up @@ -10,4 +10,4 @@ pycuda==2024.1.2
scikit-image==0.24.0
torchgeometry==0.1.2
torchvision==0.19.1
nvidia-ml-py==12.560.30
nvidia-ml-py==12.560.30 # Includes pynvml.
2 changes: 1 addition & 1 deletion runner/requirements.live-ai.txt
Original file line number Diff line number Diff line change
Expand Up @@ -22,4 +22,4 @@ PyGObject==3.50.0
onnxruntime==1.19.2
watchdog==5.0.2
aiohttp==3.10.9
nvidia-ml-py==12.560.30
nvidia-ml-py==12.560.30 # Includes pynvml.
3 changes: 1 addition & 2 deletions runner/requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -20,5 +20,4 @@ protobuf==5.27.2
bitsandbytes==0.43.3
psutil==6.0.0
PyYAML==6.0.2
nvidia-ml-py==12.560.30
pynvml==12.0.0
nvidia-ml-py==12.560.30 # Includes pynvml.

0 comments on commit 2ff0f77

Please sign in to comment.