Skip to main content

Composable sampling functions for diffusion models

Project description

Skrample 0.5.2

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.2.tar.gz (23.5 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.2-py3-none-any.whl (28.0 kB view details)

Uploaded Python 3

File details

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

File metadata

  • Download URL: skrample-0.5.2.tar.gz
  • Upload date:
  • Size: 23.5 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: uv/0.10.4 {"installer":{"name":"uv","version":"0.10.4","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.2.tar.gz
Algorithm Hash digest
SHA256 754b36eafe1dd68bab541b3a68ef47ddffd2c37f204ca6040f3a1ca9bba85ec5
MD5 f686ec8d50ae1191a52b926c5fb9b5e3
BLAKE2b-256 cd507d57c1d041a6ce58dab6b70314b2a5a9881a1245247a71e5f19d1f298628

See more details on using hashes here.

File details

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

File metadata

  • Download URL: skrample-0.5.2-py3-none-any.whl
  • Upload date:
  • Size: 28.0 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: uv/0.10.4 {"installer":{"name":"uv","version":"0.10.4","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.2-py3-none-any.whl
Algorithm Hash digest
SHA256 24ce2be7b3f1f4267fbf03c6dfafccaf229b10ff7fa068139ece1047428297f0
MD5 e64264d94ce2c9dcfe1b8aa8dab03cdf
BLAKE2b-256 50d96bd159f15ce8424f4a11e7817cc41f82bcac0a3cc1adafc12585c6131c90

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