Skip to content

Stable Diffusion XL training and inference as a cog model

Notifications You must be signed in to change notification settings

starryskyai/cog-sdxl

This branch is 28 commits behind replicate/cog-sdxl:main.

Repository files navigation

Cog-SDXL

Replicate demo and cloud API

This is an implementation of the SDXL as a Cog model. Cog packages machine learning models as standard containers.

Development

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

Basic Usage

for prediction,

cog predict -i prompt="a photo of TOK"
cog train -i input_images=@example_datasets/zeke.zip" -i use_face_detection_instead=True

Update notes

  • 2023-08-11
  • Default to 768x768 resolution training
  • Rank as argument now, default to 32
  • Now uses Swin2SR caidas/swin2SR-realworld-sr-x4-64-bsrgan-psnr as default, and will upscale + downscale to 768x768

About

Stable Diffusion XL training and inference as a cog model

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%