segmind

Documentation

Index

• Variables
• type Options
  • func WithApiKey(apiKey string) Options
  • func WithBatchSize(batchSize int) Options
  • func WithResolution(res Resolutions) Options
  • func WithS3Dir(s3dir string) Options
• type Resolutions
• type Segmind
  • func NewSegmind(model SegmindModels, opts ...Options) *Segmind
  • func (s *Segmind) RequestCreateImage(prompt string) (*string, error)
• type SegmindModels

Variables

var (
	R1024x1024 = Resolutions{Height: 1024, Width: 1024}
	R896x1152  = Resolutions{Height: 1152, Width: 896}
	R832x1216  = Resolutions{Height: 1216, Width: 832}
	R768x1344  = Resolutions{Height: 1344, Width: 768}
	R640x1536  = Resolutions{Height: 1536, Width: 640}
	R1216x832  = Resolutions{Height: 832, Width: 1216}
	R1344x768  = Resolutions{Height: 768, Width: 1344}
	R1536x640  = Resolutions{Height: 640, Width: 1536}
)

Types

type Options

type Options func(*Segmind)

func WithApiKey

func WithApiKey(apiKey string) Options

func WithBatchSize

func WithBatchSize(batchSize int) Options

func WithResolution

func WithResolution(res Resolutions) Options

func WithS3Dir

func WithS3Dir(s3dir string) Options

type Resolutions

type Resolutions struct {
	Height int `json:"height"`
	Width  int `json:"width"`
}

type Segmind

type Segmind struct {
	Model     SegmindModels
	ApiKey    string
	BatchSize int
	Width     int
	Height    int
	S3Dir     string
}

func NewSegmind

func NewSegmind(model SegmindModels, opts ...Options) *Segmind

func (*Segmind) RequestCreateImage

func (s *Segmind) RequestCreateImage(prompt string) (*string, error)

type SegmindModels

type SegmindModels string

const (
	SegmindSDAPI          SegmindModels = "https://api.segmind.com/v1/stable-diffusion-3.5-large-txt2img"
	SegmindProtovisAPI    SegmindModels = "https://api.segmind.com/v1/sdxl1.0-protovis-lightning"
	SegmindSamaritanAPI   SegmindModels = "https://api.segmind.com/v1/sdxl1.0-samaritan-3d"
	SegmindDreamshaperAPI SegmindModels = "https://api.segmind.com/v1/sdxl1.0-dreamshaper-lightning"
)