Skip to main content

Composable sampling functions for diffusion models

Project description

Skrample 0.5.1

Composable sampling functions for diffusion models

Status

Mostly complete for common models, superseding all diffusers features in quickdif

Quickstart

Fastest way to jump in is examples. The classes and functions themselves have docstrings and type hints, so it's recommended to make liberal use of your IDE or python help()

Feature Flags

  • beta-schedule -> scipy : For the Beta() schedule modifier
  • brownian-noise -> torchsde : For the Brownian() noise generator
  • cdf-schedule -> scipy : For the SigmoidCDF() schedule
  • diffusers-wrapper -> torch : For the diffusers integration module
  • pytorch -> torch : For the pytorch module
    • pytorch.noise : Custom generators
  • all : All of the above
  • dev : For running tests/

Samplers

  • Euler
  • DPM
    • 1st order, 2nd order, 3rd order
    • SDE
  • Adams/IPNDM
  • UniP & UniPC
    • N order, limited to 9 for stability
    • Custom solver via other SkrampleSampler types
  • SPC
    • Basic fully customizable midpoint corrector

Schedules

  • Linear
  • Scaled
    • uniform flag, AKA "trailing" in diffusers
  • SigmaCDF
  • ZSNR

Schedule modifiers

  • Karras
  • Exponential
  • FlowShift
  • Beta
  • Hyper

Predictors

  • Epsilon
  • Velocity / vpred
  • Flow

Noise generators

  • Random
  • Brownian
  • Offset
  • Pyramid

Integrations

Diffusers

  • Compatibility for pipelines
    • SD1
    • SDXL
    • SD3
    • Flux
  • Import from config
    • Sampler
    • Schedule
    • Predictor
  • Manage state
    • Steps
    • Higher order
    • Generators
    • Config as presented

Implementations

quickdif

My diffusers cli quickdif has full support for all major Skrample features, allowing extremely fine-grained customization.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

skrample-0.5.1.tar.gz (18.9 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

skrample-0.5.1-py3-none-any.whl (21.5 kB view details)

Uploaded Python 3

File details

Details for the file skrample-0.5.1.tar.gz.

File metadata

  • Download URL: skrample-0.5.1.tar.gz
  • Upload date:
  • Size: 18.9 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: uv/0.9.17 {"installer":{"name":"uv","version":"0.9.17","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"Arch Linux","version":null,"id":null,"libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":null}

File hashes

Hashes for skrample-0.5.1.tar.gz
Algorithm Hash digest
SHA256 ad381fffeef585de859115dfc0c8bf1209df0d397dce1343e497683845a461c2
MD5 86b014d20536f0382649aa1c5a25df8d
BLAKE2b-256 aa901c1979aad111ed9c4e92551745779b4f145508dfa362bb5727fda564584c

See more details on using hashes here.

File details

Details for the file skrample-0.5.1-py3-none-any.whl.

File metadata

  • Download URL: skrample-0.5.1-py3-none-any.whl
  • Upload date:
  • Size: 21.5 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: uv/0.9.17 {"installer":{"name":"uv","version":"0.9.17","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"Arch Linux","version":null,"id":null,"libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":null}

File hashes

Hashes for skrample-0.5.1-py3-none-any.whl
Algorithm Hash digest
SHA256 939ea847a582edce8a4ea922f31b760caae5f1119b651da7653c2318530168af
MD5 552a9c7e5e984caafcc58a5ae364ceac
BLAKE2b-256 537e40a64eb40a0f02885a7bd2b140eeb61a1139f889961272b99fdc52fc6138

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page