You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
35 lines
610 B
35 lines
610 B
//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 {
|
|
if height > 512 || width > 512 {
|
|
return stableDiffusion.GenerateImageUpscaled(
|
|
height,
|
|
width,
|
|
step,
|
|
seed,
|
|
positive_prompt,
|
|
negative_prompt,
|
|
dst,
|
|
asset_dir,
|
|
)
|
|
}
|
|
return stableDiffusion.GenerateImage(
|
|
height,
|
|
width,
|
|
mode,
|
|
step,
|
|
seed,
|
|
positive_prompt,
|
|
negative_prompt,
|
|
dst,
|
|
"",
|
|
asset_dir,
|
|
)
|
|
}
|
|
|