|
|
|
|
|
|
|
build: |
|
gpu: true |
|
cuda: "11.8" |
|
python_version: "3.9" |
|
system_packages: |
|
- "libgl1-mesa-glx" |
|
- "ffmpeg" |
|
- "libsm6" |
|
- "libxext6" |
|
- "wget" |
|
python_packages: |
|
- "diffusers<=0.25" |
|
- "torch==2.0.1" |
|
- "transformers==4.31.0" |
|
- "invisible-watermark==0.2.0" |
|
- "accelerate==0.21.0" |
|
- "pandas==2.0.3" |
|
- "torchvision==0.15.2" |
|
- "numpy==1.25.1" |
|
- "pandas==2.0.3" |
|
- "fire==0.5.0" |
|
- "opencv-python>=4.1.0.25" |
|
- "mediapipe==0.10.2" |
|
|
|
run: |
|
- curl -o /usr/local/bin/pget -L "https://github.com/replicate/pget/releases/latest/download/pget_$(uname -s)_$(uname -m)" && chmod +x /usr/local/bin/pget |
|
- 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 |
|
|
|
predict: "predict.py:Predictor" |
|
train: "train.py:train" |
|
|
