Versions in this module Expand all Collapse all v0 v0.2.1 Jun 20, 2024 Changes in this version + const COUNT + const F16 + const F32 + const I16 + const I32 + const I8 + const Q2_K + const Q3_K + const Q4_0 + const Q4_1 + const Q4_K + const Q5_0 + const Q5_1 + const Q5_K + const Q6_K + const Q8_0 + const Q8_1 + const Q8_K + var DefaultFullParams = FullParams + var DefaultOptions = Options + type CLogCallback func(level LogLevel, text string) + type CStableDiffusion interface + 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 struct + func NewCStableDiffusion(libraryPath string) (*CStableDiffusionImpl, error) + 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 Driver struct — windows/amd64 + AdapterCompatibility string + AdapterRAM int64 + Name string + func (d *Driver) Available() bool + 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 GPU struct — windows/amd64 + func NewGPU() (*GPU, error) + func (g *GPU) Cuda() *Driver + func (g *GPU) Info() []Driver + func (g *GPU) ROCm() *Driver + type Image struct + Channel uint32 + Data []byte + Height uint32 + Width uint32 + type LogLevel int + const DEBUG + const ERROR + const INFO + const WARN + 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 OutputsImageType string + const JPEG + const PNG + type RNGType int + const CUDA_RNG + const STD_DEFAULT_RNG + type SampleMethod int + const DPM2 + const DPMPP2M + const DPMPP2Mv2 + const DPMPP2S_A + const EULER + const EULER_A + const HEUN + const LCM + const N_SAMPLE_METHODS + type Schedule int + const DEFAULT + const DISCRETE + const KARRAS + const N_SCHEDULES + type WType int