chore: switch to flux pro v1.1 for image generation

davidramiro · Nov 26, 2024 · 910fc13 · 910fc13
1 parent ad2dddc
commit 910fc13
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/config.sample.toml b/config.sample.toml
@@ -23,5 +23,5 @@ You are HSBot, a helpful assistant.
 
 [fal]
 api_key = "4242:1234"
-flux_url = "https://fal.run/fal-ai/flux-pro"
+flux_url = "https://fal.run/fal-ai/flux-pro/v1.1"
 whisper_url = "https://fal.run/fal-ai/whisper"
diff --git a/internal/adapters/generator/fal.go b/internal/adapters/generator/fal.go
@@ -25,6 +25,7 @@ func NewFALGenerator(apiURL, apiKey string) *FALGenerator {
 type imageRequest struct {
 	Prompt              string `json:"prompt"`
 	EnableSafetyChecker bool   `json:"enable_safety_checker"`
+	SafetyTolerance     string `json:"safety_tolerance"`
 	ImageSize           string `json:"image_size"`
 }
 
@@ -41,6 +42,7 @@ func (f *FALGenerator) GenerateFromPrompt(ctx context.Context, prompt string) (s
 		Prompt:              prompt,
 		EnableSafetyChecker: false,
 		ImageSize:           "square",
+		SafetyTolerance:     "5",
 	}
 
 	payloadBuf := new(bytes.Buffer)