mirror of
https://github.com/mudler/LocalAI.git
synced 2024-12-25 15:21:04 +00:00
24 lines
420 B
Go
24 lines
420 B
Go
|
//go:build stablediffusion
|
||
|
|
// +build stablediffusion
|
||
|
|
|
||
|
|
package stablediffusion
|
||
|
|
|
||
|
|
import (
|
||
|
|
stableDiffusion "github.com/mudler/go-stable-diffusion"
|
||
|
|
)
|
||
|
|
|
||
|
|
func GenerateImage(height, width, mode, step, seed int, positive_prompt, negative_prompt, dst, asset_dir string) error {
|
||
|
|
return stableDiffusion.GenerateImage(
|
||
|
|
height,
|
||
|
|
width,
|
||
|
|
mode,
|
||
|
|
step,
|
||
|
|
seed,
|
||
|
|
positive_prompt,
|
||
|
|
negative_prompt,
|
||
|
|
dst,
|
||
|
|
"",
|
||
|
|
asset_dir,
|
||
|
|
)
|
||
|
|
}
|