stablediffusion

type ContainerRunner interface {
	// Run runs a container with the given image and arguments to completion,
	// and returns its combined output as a byte string.
	Run(ctx context.Context, image string, argv []string) ([]byte, error)
}

type XL struct {
	// contains filtered or unexported fields
}

type XLImage struct {
	Prompt *XLPrompt
	// contains filtered or unexported fields
}

type XLPrompt struct {
	// Query is the text query to generate the image with.
	Query string

	// AllowCPUOffload is whether to allow offloading parts of the model to CPU.
	AllowCPUOffload bool

	// UseRefiner is whether to use the refiner model after the base model.
	// This takes more VRAM and more time but produces a better image.
	UseRefiner bool

	// NoiseFraction is the fraction of noise to seed the image with.
	// Must be between 0.0 and 1.0 inclusively.
	NoiseFraction float64

	// Steps is the number of diffusion steps to run for the base and refiner
	// models. More steps generally means sharper results but more time to
	// generate the image. A reasonable value is between 30 and 50.
	Steps int

	// Warm controls whether the image will be generated while the model is
	// warm. This will double the running time, as the image will still be
	// generated with a cold model first.
	Warm bool
}