diff --git a/cog.yaml b/cog.yaml index a02f5f1..d4b1520 100644 --- a/cog.yaml +++ b/cog.yaml @@ -3,7 +3,12 @@ build: gpu: true - python_version: "3.11" + python_version: "3.10" + system_packages: + - "ffmpeg" + - "libsm6" + - "libxext6" + - "libgl1" python_packages: - "torch==2.3.0" - "torchvision==0.18.0"