forked from LiveCarta/ContentGeneration
Run video generation model from python instead of calling subprocess
This commit is contained in:
@@ -3,6 +3,7 @@ FROM nvidia/cuda:12.4.1-cudnn-devel-ubuntu22.04
|
||||
ENV DEBIAN_FRONTEND=noninteractive \
|
||||
PYTHONUNBUFFERED=1 \
|
||||
UV_EXTRA_INDEX_URL=https://download.pytorch.org/whl/cu121 \
|
||||
UV_NO_SYNC=true \
|
||||
PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True,max_split_size_mb:128
|
||||
|
||||
# Base OS tools + media stack + Python toolchain.
|
||||
|
||||
Reference in New Issue
Block a user