Compare commits
2 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| a213c18263 | |||
| a744cdc335 |
@@ -42,3 +42,22 @@ asked for "a new ai image interface as opposed to llm".
|
||||
callers (additive fields/options).
|
||||
- No health/failover for image models yet; if needed it can be added as a
|
||||
separate chain type rather than retrofitting the chat chain.
|
||||
|
||||
## Update — optional per-request settings
|
||||
|
||||
`Request` gained additive optional overrides — `Steps *int`, `CFGScale *float64`,
|
||||
`NegativePrompt string`, `Sampler string`, `Seed *int64` — with mirror options
|
||||
(`WithSteps`, …). nil/"" means "leave the backend's per-model default", so the v1
|
||||
contract is unchanged for callers that don't set them. `provider/llamaswap`
|
||||
forwards them to sd-server as `steps`/`cfg_scale`/`negative_prompt`/`sample_method`/
|
||||
`seed` (omitempty). This realizes the "seeds/steps … additive fields" note above;
|
||||
img2img/masks/streaming remain deferred.
|
||||
|
||||
## Update — A1111 txt2img endpoint (seed support)
|
||||
|
||||
`provider/llamaswap` now POSTs to sd-server's **`/sdapi/v1/txt2img`** (A1111)
|
||||
instead of the OpenAI `/v1/images/generations`. That OpenAI endpoint **ignores
|
||||
`seed`** on the stable-diffusion.cpp build we run — every render of a prompt is
|
||||
byte-identical, so a batch of N collapses to one image. `/sdapi/v1/txt2img`
|
||||
honours `seed`, restoring real per-render variety. llama-swap still routes by
|
||||
the `model` field in the body; `Size` is split into `width`/`height`.
|
||||
|
||||
@@ -38,6 +38,29 @@ type Request struct {
|
||||
// Size is the requested resolution, e.g. "512x512" or "1024x1024";
|
||||
// "" = provider default.
|
||||
Size string
|
||||
|
||||
// The fields below are optional per-request overrides. Their zero value
|
||||
// (nil pointer or empty string) means "leave the backend's own default" —
|
||||
// for stable-diffusion.cpp that is the per-model default baked into the
|
||||
// llama-swap launch flags. A caller overrides only what it explicitly sets.
|
||||
|
||||
// Steps is the number of diffusion steps; nil = backend default.
|
||||
Steps *int
|
||||
|
||||
// CFGScale is the classifier-free-guidance scale; nil = backend default.
|
||||
// Architecture-sensitive (SDXL likes ~7, Flux wants 1), so prefer leaving
|
||||
// it nil unless the caller knows the target model.
|
||||
CFGScale *float64
|
||||
|
||||
// NegativePrompt steers generation away from concepts; "" = none.
|
||||
NegativePrompt string
|
||||
|
||||
// Sampler selects the sampling method (e.g. "euler", "euler_a");
|
||||
// "" = backend default.
|
||||
Sampler string
|
||||
|
||||
// Seed fixes the RNG seed for reproducible output; nil = random.
|
||||
Seed *int64
|
||||
}
|
||||
|
||||
// Result is the canonical image-generation result.
|
||||
@@ -60,6 +83,21 @@ func WithN(n int) Option { return func(r *Request) { r.N = n } }
|
||||
// WithSize sets the requested resolution (e.g. "1024x1024").
|
||||
func WithSize(size string) Option { return func(r *Request) { r.Size = size } }
|
||||
|
||||
// WithSteps overrides the number of diffusion steps.
|
||||
func WithSteps(n int) Option { return func(r *Request) { r.Steps = &n } }
|
||||
|
||||
// WithCFGScale overrides the classifier-free-guidance scale.
|
||||
func WithCFGScale(s float64) Option { return func(r *Request) { r.CFGScale = &s } }
|
||||
|
||||
// WithNegativePrompt sets a negative prompt.
|
||||
func WithNegativePrompt(s string) Option { return func(r *Request) { r.NegativePrompt = s } }
|
||||
|
||||
// WithSampler overrides the sampling method (e.g. "euler", "euler_a").
|
||||
func WithSampler(s string) Option { return func(r *Request) { r.Sampler = s } }
|
||||
|
||||
// WithSeed fixes the RNG seed for reproducible output.
|
||||
func WithSeed(seed int64) Option { return func(r *Request) { r.Seed = &seed } }
|
||||
|
||||
// Apply returns a copy of the request with all options applied. Providers call
|
||||
// this once at the top of Generate.
|
||||
func (r Request) Apply(opts ...Option) Request {
|
||||
|
||||
+64
-34
@@ -5,6 +5,7 @@ import (
|
||||
"encoding/base64"
|
||||
"fmt"
|
||||
"net/http"
|
||||
"strconv"
|
||||
"strings"
|
||||
|
||||
"gitea.stevedudenhoeffer.com/steve/majordomo/imagegen"
|
||||
@@ -27,25 +28,32 @@ type imageModel struct {
|
||||
id string
|
||||
}
|
||||
|
||||
// imageRequest is the OpenAI /v1/images/generations request shape. We always
|
||||
// request b64_json so the bytes come back inline (no second fetch).
|
||||
type imageRequest struct {
|
||||
Model string `json:"model"`
|
||||
Prompt string `json:"prompt"`
|
||||
N int `json:"n,omitempty"`
|
||||
Size string `json:"size,omitempty"`
|
||||
ResponseFormat string `json:"response_format"`
|
||||
// txt2imgRequest is the stable-diffusion.cpp sd-server A1111 request shape
|
||||
// (POST /sdapi/v1/txt2img). We use this endpoint rather than the OpenAI
|
||||
// /v1/images/generations one because that endpoint IGNORES `seed` on this
|
||||
// sd-server build — every render of a given prompt comes back byte-identical,
|
||||
// so a batch of N collapses to one image. /sdapi/v1/txt2img honours `seed`,
|
||||
// giving real variety. llama-swap still routes by the `model` field in the
|
||||
// body. Optional fields are pointers/omitempty so an unset value falls back to
|
||||
// the model's baked default (the per-model --steps/--cfg-scale/etc. flags).
|
||||
type txt2imgRequest struct {
|
||||
Model string `json:"model"`
|
||||
Prompt string `json:"prompt"`
|
||||
NegativePrompt string `json:"negative_prompt,omitempty"`
|
||||
Seed *int64 `json:"seed,omitempty"`
|
||||
Steps *int `json:"steps,omitempty"`
|
||||
CFGScale *float64 `json:"cfg_scale,omitempty"`
|
||||
Width *int `json:"width,omitempty"`
|
||||
Height *int `json:"height,omitempty"`
|
||||
SampleMethod string `json:"sample_method,omitempty"`
|
||||
BatchCount int `json:"batch_count,omitempty"`
|
||||
}
|
||||
|
||||
type imageResponse struct {
|
||||
Created int64 `json:"created"`
|
||||
Data []struct {
|
||||
B64JSON string `json:"b64_json"`
|
||||
URL string `json:"url"`
|
||||
} `json:"data"`
|
||||
type txt2imgResponse struct {
|
||||
Images []string `json:"images"`
|
||||
}
|
||||
|
||||
// Generate implements imagegen.Model via POST {base}/v1/images/generations.
|
||||
// Generate implements imagegen.Model via POST {base}/sdapi/v1/txt2img.
|
||||
func (m *imageModel) Generate(ctx context.Context, req imagegen.Request, opts ...imagegen.Option) (*imagegen.Result, error) {
|
||||
req = req.Apply(opts...)
|
||||
if strings.TrimSpace(req.Prompt) == "" {
|
||||
@@ -55,32 +63,35 @@ func (m *imageModel) Generate(ctx context.Context, req imagegen.Request, opts ..
|
||||
return nil, fmt.Errorf("%w: image count N must be >= 0, got %d", llm.ErrUnsupported, req.N)
|
||||
}
|
||||
|
||||
wire := imageRequest{
|
||||
Model: m.id,
|
||||
Prompt: req.Prompt,
|
||||
N: req.N,
|
||||
Size: req.Size,
|
||||
ResponseFormat: "b64_json",
|
||||
width, height, err := parseSize(req.Size)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("%w: %v", llm.ErrUnsupported, err)
|
||||
}
|
||||
|
||||
var resp imageResponse
|
||||
if err := m.p.doJSON(ctx, http.MethodPost, "/v1/images/generations", m.id, &wire, &resp); err != nil {
|
||||
wire := txt2imgRequest{
|
||||
Model: m.id,
|
||||
Prompt: req.Prompt,
|
||||
NegativePrompt: req.NegativePrompt,
|
||||
Seed: req.Seed,
|
||||
Steps: req.Steps,
|
||||
CFGScale: req.CFGScale,
|
||||
Width: width,
|
||||
Height: height,
|
||||
SampleMethod: req.Sampler,
|
||||
BatchCount: req.N,
|
||||
}
|
||||
|
||||
var resp txt2imgResponse
|
||||
if err := m.p.doJSON(ctx, http.MethodPost, "/sdapi/v1/txt2img", m.id, &wire, &resp); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
out := &imagegen.Result{Raw: &resp}
|
||||
for i, d := range resp.Data {
|
||||
if d.B64JSON == "" {
|
||||
// Why error rather than skip: a url-only entry means the backend
|
||||
// ignored response_format; we don't fetch remote content (mirrors
|
||||
// llm.ImagePart's bytes-only contract), so surface it.
|
||||
return nil, &llm.APIError{
|
||||
Provider: m.p.name,
|
||||
Model: m.id,
|
||||
Message: fmt.Sprintf("image %d returned no inline b64_json data", i),
|
||||
}
|
||||
for i, b64 := range resp.Images {
|
||||
if b64 == "" {
|
||||
continue
|
||||
}
|
||||
raw, err := base64.StdEncoding.DecodeString(d.B64JSON)
|
||||
raw, err := base64.StdEncoding.DecodeString(b64)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("llama-swap: decode image %d: %w", i, err)
|
||||
}
|
||||
@@ -96,6 +107,25 @@ func (m *imageModel) Generate(ctx context.Context, req imagegen.Request, opts ..
|
||||
return out, nil
|
||||
}
|
||||
|
||||
// parseSize splits a "WxH" string into width/height pointers. "" yields
|
||||
// (nil, nil) so the model's own default resolution applies.
|
||||
func parseSize(size string) (*int, *int, error) {
|
||||
size = strings.TrimSpace(size)
|
||||
if size == "" {
|
||||
return nil, nil, nil
|
||||
}
|
||||
parts := strings.SplitN(strings.ToLower(size), "x", 2)
|
||||
if len(parts) != 2 {
|
||||
return nil, nil, fmt.Errorf("invalid size %q (want WxH)", size)
|
||||
}
|
||||
w, err1 := strconv.Atoi(strings.TrimSpace(parts[0]))
|
||||
h, err2 := strconv.Atoi(strings.TrimSpace(parts[1]))
|
||||
if err1 != nil || err2 != nil || w <= 0 || h <= 0 {
|
||||
return nil, nil, fmt.Errorf("invalid size %q (want WxH)", size)
|
||||
}
|
||||
return &w, &h, nil
|
||||
}
|
||||
|
||||
// sniffImageMIME identifies the image format from its leading bytes, defaulting
|
||||
// to image/png (stable-diffusion.cpp emits PNG) when detection is inconclusive.
|
||||
func sniffImageMIME(data []byte) string {
|
||||
|
||||
@@ -166,11 +166,11 @@ func TestRunningRaw(t *testing.T) {
|
||||
func TestImageGenerate(t *testing.T) {
|
||||
var gotBody map[string]any
|
||||
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
if r.URL.Path != "/v1/images/generations" {
|
||||
if r.URL.Path != "/sdapi/v1/txt2img" {
|
||||
t.Errorf("path = %q", r.URL.Path)
|
||||
}
|
||||
_ = json.NewDecoder(r.Body).Decode(&gotBody)
|
||||
_, _ = w.Write([]byte(`{"created":1,"data":[{"b64_json":"` + onePixelPNG + `"}]}`))
|
||||
_, _ = w.Write([]byte(`{"images":["` + onePixelPNG + `"]}`))
|
||||
}))
|
||||
defer srv.Close()
|
||||
|
||||
@@ -192,12 +192,51 @@ func TestImageGenerate(t *testing.T) {
|
||||
if len(res.Images[0].Data) == 0 {
|
||||
t.Error("decoded image has no bytes")
|
||||
}
|
||||
// response_format must be forced to b64_json, and options applied.
|
||||
if gotBody["response_format"] != "b64_json" {
|
||||
t.Errorf("response_format = %v, want b64_json", gotBody["response_format"])
|
||||
// Size is split into width/height ints for the A1111 endpoint.
|
||||
if gotBody["width"] != float64(512) || gotBody["height"] != float64(512) {
|
||||
t.Errorf("width/height = %v/%v, want 512/512", gotBody["width"], gotBody["height"])
|
||||
}
|
||||
if gotBody["size"] != "512x512" {
|
||||
t.Errorf("size = %v, want 512x512", gotBody["size"])
|
||||
}
|
||||
|
||||
func TestImageGenerateSettings(t *testing.T) {
|
||||
var gotBody map[string]any
|
||||
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
_ = json.NewDecoder(r.Body).Decode(&gotBody)
|
||||
_, _ = w.Write([]byte(`{"images":["` + onePixelPNG + `"]}`))
|
||||
}))
|
||||
defer srv.Close()
|
||||
|
||||
p := New(WithBaseURL(srv.URL), WithHTTPClient(srv.Client()))
|
||||
im, _ := p.ImageModel("sd")
|
||||
|
||||
// Unset overrides must be omitted entirely so sd-server keeps its own
|
||||
// per-model defaults.
|
||||
if _, err := im.Generate(context.Background(), imagegen.Request{Prompt: "x"}); err != nil {
|
||||
t.Fatalf("Generate: %v", err)
|
||||
}
|
||||
for _, k := range []string{"steps", "cfg_scale", "negative_prompt", "sample_method", "seed"} {
|
||||
if v, ok := gotBody[k]; ok {
|
||||
t.Errorf("unset request sent %q = %v, want omitted", k, v)
|
||||
}
|
||||
}
|
||||
|
||||
// Set overrides are forwarded with the sd-server-friendly field names.
|
||||
gotBody = nil
|
||||
_, err := im.Generate(context.Background(), imagegen.Request{Prompt: "x"},
|
||||
imagegen.WithSteps(8),
|
||||
imagegen.WithCFGScale(3.5),
|
||||
imagegen.WithNegativePrompt("blurry"),
|
||||
imagegen.WithSampler("euler"),
|
||||
imagegen.WithSeed(42),
|
||||
)
|
||||
if err != nil {
|
||||
t.Fatalf("Generate: %v", err)
|
||||
}
|
||||
want := map[string]any{"steps": float64(8), "cfg_scale": 3.5, "negative_prompt": "blurry", "sample_method": "euler", "seed": float64(42)}
|
||||
for k, w := range want {
|
||||
if gotBody[k] != w {
|
||||
t.Errorf("%s = %v, want %v", k, gotBody[k], w)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user