metrosir
/

sd

Inference Endpoints

Model card Files Files and versions Community

sd / cog.yaml

zhengqilin

add init

1976a91 about 1 year ago

raw history blame contribute delete

No virus

2.16 kB

	# Configuration for Cog ⚙️
	# https://replicate.com/docs/guides/push-a-model
	# prerequisite：https://docs.docker.com/engine/install/ubuntu/#set-up-the-repository `dockerd` to start docker
	# Reference: https://github.com/replicate/cog/blob/main/docs/yaml.md
	# !!!! recommend 60G disk space for cog docker

	build:
	# set to true if your model requires a GPU
	gpu: true

	# a list of ubuntu apt packages to install
	system_packages:
	- "libgl1-mesa-glx"
	- "libglib2.0-0"

	# python version in the form '3.8' or '3.8.12'
	python_version: "3.10.4"

	# a list of packages in the format <package-name>==<version>
	python_packages:
	- blendmodes==2022
	- transformers==4.25.1
	- accelerate==0.12.0
	- basicsr==1.4.2
	- gfpgan==1.3.8
	- gradio==3.16.2
	- numpy==1.23.3
	- Pillow==9.4.0
	- realesrgan==0.3.0
	# - torch==1.13.1+cu117
	# - --extra-index-url https://download.pytorch.org/whl/cu117
	# - torchvision==0.14.1+cu117
	# - --extra-index-url https://download.pytorch.org/whl/cu117
	- omegaconf==2.2.3
	- pytorch_lightning==1.7.6
	- scikit-image==0.19.2
	- fonts
	- font-roboto
	- timm==0.6.7
	- piexif==1.1.3
	- einops==0.4.1
	- jsonmerge==1.8.0
	- clean-fid==0.1.29
	- resize-right==0.0.2
	- torchdiffeq==0.2.3
	- kornia==0.6.7
	- lark==1.1.2
	- inflection==0.5.1
	- GitPython==3.1.27
	- torchsde==0.2.5
	- safetensors==0.2.7
	- httpcore<=0.15
	- fastapi==0.90.1
	# - open_clip_torch
	- git+https://github.com/mlfoundations/open_clip.git@bb6e834e9c70d9c27d0dc3ecedeebeaeb1ffad6b
	- git+https://github.com/openai/CLIP.git@d50d76daa670286dd6cacf3bcd80b5e4823fc8e1

	# commands run after the environment is setup
	run:
	- "pip3 install torch==1.13.1+cu117 --extra-index-url https://download.pytorch.org/whl/cu117"
	- "pip3 install torchvision==0.14.1+cu117 --extra-index-url https://download.pytorch.org/whl/cu117"
	- "echo env is ready!"

	# https://replicate.com/wolverinn/chill_watcher
	image: "r8.im/wolverinn/chill_watcher"

	# predict.py defines how predictions are run on your model
	predict: "predict.py:Predictor"