sd

package module

Go to main page

Versions in this module

v0

v0.2.0

Jan 21, 2024

v0.1.3

Jan 16, 2024

v0.1.2

Jan 11, 2024

v0.1.1

Jan 10, 2024

Changes in this version

+ type Driver struct — windows/amd64

+ AdapterCompatibility string

+ AdapterRAM int64

+ Name string

+ func (d *Driver) Available() bool

+ type GPU struct — windows/amd64

+ func NewGPU() (*GPU, error)

+ func (g *GPU) Cuda() *Driver

+ func (g *GPU) Info() []Driver

+ func (g *GPU) ROCm() *Driver

v0.1.0

Jan 6, 2024

Changes in this version

+ const COUNT

+ const I16

+ const I32

+ const I8

+ const Q2_K

+ const Q3_K

+ const Q4_K

+ const Q5_K

+ const Q6_K

+ const Q8_1

+ const Q8_K

+ var DefaultFullParams = FullParams

+ var DefaultOptions = Options

+ type CLogCallback func(level LogLevel, text string)

type CStableDiffusion

+ Close func() error

+ FreeCtx func(ctx *CStableDiffusionCtx)

+ FreeUpscalerCtx func(ctx *CUpScalerCtx)

+ GetSystemInfo func() string

+ ImagePredictImage func(ctx *CStableDiffusionCtx, img Image, prompt string, negativePrompt string, ...) []Image

+ NewCtx func(modelPath string, vaePath string, taesdPath string, loraModelDir string, ...) *CStableDiffusionCtx

+ NewUpscalerCtx func(esrganPath string, nThreads int, wType WType) *CUpScalerCtx

+ PredictImage func(ctx *CStableDiffusionCtx, prompt string, negativePrompt string, clipSkip int, ...) []Image

+ SetLogCallBack func(cb CLogCallback)

+ UpscaleImage func(ctx *CUpScalerCtx, img Image, upscaleFactor uint32) Image

+ type CStableDiffusionCtx struct

type CStableDiffusionImpl

+ func (c *CStableDiffusionImpl) Close() error

+ func (c *CStableDiffusionImpl) FreeCtx(ctx *CStableDiffusionCtx)

+ func (c *CStableDiffusionImpl) FreeUpscalerCtx(ctx *CUpScalerCtx)

+ func (c *CStableDiffusionImpl) GetSystemInfo() string

+ func (c *CStableDiffusionImpl) ImagePredictImage(ctx *CStableDiffusionCtx, img Image, prompt string, negativePrompt string, ...) []Image

+ func (c *CStableDiffusionImpl) NewCtx(modelPath string, vaePath string, taesdPath string, loraModelDir string, ...) *CStableDiffusionCtx

+ func (c *CStableDiffusionImpl) NewUpscalerCtx(esrganPath string, nThreads int, wType WType) *CUpScalerCtx

+ func (c *CStableDiffusionImpl) PredictImage(ctx *CStableDiffusionCtx, prompt string, negativePrompt string, clipSkip int, ...) []Image

+ func (c *CStableDiffusionImpl) SetLogCallBack(cb CLogCallback)

+ func (c *CStableDiffusionImpl) UpscaleImage(ctx *CUpScalerCtx, img Image, upscaleFactor uint32) Image

+ type CUpScalerCtx struct

+ type FullParams struct

+ BatchCount int

+ CfgScale float32

+ ClipSkip int

+ Height int

+ NegativePrompt string

+ OutputsImageType OutputsImageType

+ SampleMethod SampleMethod

+ SampleSteps int

+ Seed int64

+ Strength float32

+ Width int

+ type Image struct

+ Channel uint32

+ Data []byte

+ Height uint32

+ Width uint32

+ type LogLevel int

+ type Model struct

+ func NewAutoModel(options Options) (*Model, error)

+ func NewModel(dylibPath string, options Options) (*Model, error)

+ func (sd *Model) Close() error

+ func (sd *Model) ImagePredict(reader io.Reader, prompt string, params FullParams, writer []io.Writer) error

+ func (sd *Model) LoadFromFile(path string) error

+ func (sd *Model) Predict(prompt string, params FullParams, writer []io.Writer) error

+ func (sd *Model) SetLogCallback(cb CLogCallback)

+ func (sd *Model) SetOptions(options Options)

+ func (sd *Model) UpscaleImage(reader io.Reader, esrganPath string, upscaleFactor uint32, writer io.Writer) error

+ type Options struct

+ FreeParamsImmediately bool

+ GpuEnable bool

+ LoraModelDir string

+ RngType RNGType

+ Schedule Schedule

+ TaesdPath string

+ Threads int

+ VaeDecodeOnly bool

+ VaePath string

+ VaeTiling bool

+ Wtype WType

+ type WType int

v0.0.6

Dec 25, 2023

Changes in this version

+ const DEBUG

+ const DEFAULT

+ const DISCRETE

+ const DPM2

+ const DPMPP2M

+ const DPMPP2Mv2

+ const DPMPP2S_A

+ const ERROR

+ const EULER

+ const EULER_A

+ const F16

+ const F32

+ const HEUN

+ const INFO

+ const KARRAS

+ const LCM

+ const N_SAMPLE_METHODS

+ const N_SCHEDULES

+ const Q4_0

+ const Q4_1

+ const Q5_0

+ const Q5_1

+ const Q8_0

+ const T_DEFAULT

+ const WARN

+ var DefaultStableDiffusionOptions = StableDiffusionOptions

+ type CStableDiffusion interface

+ StableDiffusionFree func(ctx *StableDiffusionCtx)

+ StableDiffusionFreeBuffer func(buffer uintptr)

+ StableDiffusionFreeFullParams func(params *StableDiffusionFullParams)

+ StableDiffusionFullDefaultParamsRef func() *StableDiffusionFullParams

+ StableDiffusionFullParamsSetBatchCount func(params *StableDiffusionFullParams, batchCount int)

+ StableDiffusionFullParamsSetCfgScale func(params *StableDiffusionFullParams, cfgScale float32)

+ StableDiffusionFullParamsSetHeight func(params *StableDiffusionFullParams, height int)

+ StableDiffusionFullParamsSetNegativePrompt func(params *StableDiffusionFullParams, negativePrompt string)

+ StableDiffusionFullParamsSetSampleMethod func(params *StableDiffusionFullParams, sampleMethod SampleMethod)

+ StableDiffusionFullParamsSetSampleSteps func(params *StableDiffusionFullParams, sampleSteps int)

+ StableDiffusionFullParamsSetSeed func(params *StableDiffusionFullParams, seed int64)

+ StableDiffusionFullParamsSetStrength func(params *StableDiffusionFullParams, strength float32)

+ StableDiffusionFullParamsSetWidth func(params *StableDiffusionFullParams, width int)

+ StableDiffusionGetSystemInfo func() string

+ StableDiffusionImagePredictImage func(ctx *StableDiffusionCtx, params *StableDiffusionFullParams, initImage []byte, ...) []byte

+ StableDiffusionInit func(nThreads int, vaeDecodeOnly bool, taesdPath string, freeParamsImmediately bool, ...) *StableDiffusionCtx

+ StableDiffusionLoadFromFile func(ctx *StableDiffusionCtx, filePath string, vaePath string, wtype GgmlType, ...)

+ StableDiffusionPredictImage func(ctx *StableDiffusionCtx, params *StableDiffusionFullParams, prompt string) []byte

+ StableDiffusionSetLogLevel func(level SDLogLevel)

+ func NewCStableDiffusion(libraryPath string) (CStableDiffusion, error)

+ type CStableDiffusionImpl struct

+ func (c *CStableDiffusionImpl) StableDiffusionFree(ctx *StableDiffusionCtx)

+ func (c *CStableDiffusionImpl) StableDiffusionFreeBuffer(buffer uintptr)

+ func (c *CStableDiffusionImpl) StableDiffusionFreeFullParams(params *StableDiffusionFullParams)

+ func (c *CStableDiffusionImpl) StableDiffusionFullDefaultParamsRef() *StableDiffusionFullParams

+ func (c *CStableDiffusionImpl) StableDiffusionFullParamsSetBatchCount(params *StableDiffusionFullParams, batchCount int)

+ func (c *CStableDiffusionImpl) StableDiffusionFullParamsSetCfgScale(params *StableDiffusionFullParams, cfgScale float32)

+ func (c *CStableDiffusionImpl) StableDiffusionFullParamsSetHeight(params *StableDiffusionFullParams, height int)

+ func (c *CStableDiffusionImpl) StableDiffusionFullParamsSetNegativePrompt(params *StableDiffusionFullParams, negativePrompt string)

+ func (c *CStableDiffusionImpl) StableDiffusionFullParamsSetSampleMethod(params *StableDiffusionFullParams, sampleMethod SampleMethod)

+ func (c *CStableDiffusionImpl) StableDiffusionFullParamsSetSampleSteps(params *StableDiffusionFullParams, sampleSteps int)

+ func (c *CStableDiffusionImpl) StableDiffusionFullParamsSetSeed(params *StableDiffusionFullParams, seed int64)

+ func (c *CStableDiffusionImpl) StableDiffusionFullParamsSetStrength(params *StableDiffusionFullParams, strength float32)

+ func (c *CStableDiffusionImpl) StableDiffusionFullParamsSetWidth(params *StableDiffusionFullParams, width int)

+ func (c *CStableDiffusionImpl) StableDiffusionGetSystemInfo() string

+ func (c *CStableDiffusionImpl) StableDiffusionImagePredictImage(ctx *StableDiffusionCtx, params *StableDiffusionFullParams, initImage []byte, ...) []byte

+ func (c *CStableDiffusionImpl) StableDiffusionInit(nThreads int, vaeDecodeOnly bool, taesdPath string, freeParamsImmediately bool, ...) *StableDiffusionCtx

+ func (c *CStableDiffusionImpl) StableDiffusionLoadFromFile(ctx *StableDiffusionCtx, filePath string, vaePath string, wtype GgmlType, ...)

+ func (c *CStableDiffusionImpl) StableDiffusionPredictImage(ctx *StableDiffusionCtx, params *StableDiffusionFullParams, prompt string) []byte

+ func (c *CStableDiffusionImpl) StableDiffusionSetLogLevel(level SDLogLevel)

+ type GgmlType string

+ type OutputsImageType string

+ const JPEG

+ const PNG

+ type RNGType string

+ const CUDA_RNG

+ const STD_DEFAULT_RNG

+ type SDLogLevel string

+ type SampleMethod string

+ type Schedule string

+ type StableDiffusionCtx struct

+ type StableDiffusionFullParams struct

+ type StableDiffusionModel struct

+ func NewStableDiffusionAutoModel(options StableDiffusionOptions) (*StableDiffusionModel, error)

+ func NewStableDiffusionModel(dylibPath string, options StableDiffusionOptions) (*StableDiffusionModel, error)

+ func (sd *StableDiffusionModel) Close() error

+ func (sd *StableDiffusionModel) ImagePredict(reader io.Reader, prompt string, writer io.Writer) error

+ func (sd *StableDiffusionModel) LoadFromFile(path string) error

+ func (sd *StableDiffusionModel) Predict(prompt string, writer []io.Writer) error

+ func (sd *StableDiffusionModel) SetOptions(options StableDiffusionOptions)

+ type StableDiffusionOptions struct

+ BatchCount int

+ CfgScale float32

+ FreeParamsImmediately bool

+ GpuEnable bool

+ Height int

+ LoraModelDir string

+ NegativePrompt string

+ OutputsImageType OutputsImageType

+ RngType RNGType

+ SampleMethod SampleMethod

+ SampleSteps int

+ Schedule Schedule

+ Seed int64

+ Strength float32

+ TaesdPath string

+ Threads int

+ VaeDecodeOnly bool

+ VaePath string

+ WType GgmlType

+ Width int

v0.0.5-beta1

Dec 12, 2023

v0.0.4-beta1

Nov 26, 2023

v0.0.3-beta1

Nov 24, 2023

v0.0.2-beta1

Sep 26, 2023

v0.0.1-beta1

Sep 25, 2023

?	: This menu
/	: Search site
f or F	: Jump to
y or Y	: Canonical URL