Update cog.yaml

This commit is contained in:
Gene Kogan
2024-07-16 03:44:35 -07:00
committed by GitHub
parent 7d8c845765
commit 2f5eaeba7a
+2 -7
View File
@@ -6,16 +6,11 @@ build:
cuda: "12.1"
python_version: "3.11"
system_packages:
- "libcudnn8=8.9.0.131-1+cuda12.1"
- "libcudnn8-dev=8.9.0.131-1+cuda12.1"
- "ffmpeg"
python_requirements: requirements.txt
run:
- apt-mark showhold
- apt-get update -y -qq
- apt-get upgrade -y --allow-change-held-packages
- apt-get install -y --allow-change-held-packages ffmpeg libgl1-mesa-glx libegl1-mesa-dev libsm6 libxext6 libglib2.0-0
- wget http://thegiflibrary.tumblr.com/post/11565547760 -O face_landmarker_v2_with_blendshapes.task -q https://storage.googleapis.com/mediapipe-models/face_landmarker/face_landmarker/float16/1/face_landmarker.task
- wget https://storage.googleapis.com/mediapipe-models/face_landmarker/face_landmarker/float16/1/face_landmarker.task -O face_landmarker_v2_with_blendshapes.task
predict: "predict.py:Predictor"
image: "r8.im/edenartlab/sdxl-lora-trainer"