Cog-SDXL-LCM

This is an implementation of Stability AI's SDXL as a Cog model with LCM LoRA for faster inference.

Development

Follow the model pushing guide to push your own fork of SDXL to Replicate.

Download weights first to apply the LCM-LoRA for SDXL

cog run python script/download_weights.py

Then for predictions,

cog predict -i prompt="a photo of TOK"

cog train -i input_images=@example_datasets/__data.zip -i use_face_detection_instead=True

cog run -p 5000 python -m cog.server.http

Name		Name	Last commit message	Last commit date
Latest commit History 52 Commits
example_datasets		example_datasets
feature-extractor		feature-extractor
script		script
tests		tests
.dockerignore		.dockerignore
.gitignore		.gitignore
README.md		README.md
cog.yaml		cog.yaml
dataset_and_utils.py		dataset_and_utils.py
no_init.py		no_init.py
predict.py		predict.py
preprocess.py		preprocess.py
samples.py		samples.py
train.py		train.py
trainer_pti.py		trainer_pti.py
weights.py		weights.py