update docs

landing-ai · Oct 21, 2024 · ce3c613 · ce3c613
1 parent b5e548f
commit ce3c613
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/docs/flux1.md b/docs/flux1.md
@@ -19,8 +19,8 @@ Below is an example of how to create and use a `Flux1Config` object:
 from vision_agent_tools.models.flux1 import Flux1Config
 
 config = Flux1Config(
-    height=1024,
-    width=1024,
+    height=512,
+    width=512,
     num_inference_steps=28,
     guidance_scale=3.5,
     num_images_per_prompt=1,
@@ -29,8 +29,8 @@ config = Flux1Config(
 )
 ```
 
-- height: The height in pixels of the generated image. Defaults to 1024 for best results.
-- width: The width in pixels of the generated image. Defaults to 1024 for best results.
+- height: The height in pixels of the generated image. Defaults to 512 for best results.
+- width: The width in pixels of the generated image. Defaults to 512 for best results.
 - num_inference_steps: The number of inference steps to perform. Defaults to 28.
 - guidance_scale: Guidance scale as defined in Classifier-Free Diffusion Guidance. Defaults to 3.5.
 - num_images_per_prompt: The number of images to generate per prompt. Defaults to 1.

diff --git a/vision_agent_tools/models/flux1.py b/vision_agent_tools/models/flux1.py
@@ -108,10 +108,10 @@ def __call__(
             - config (Flux1Config):
                 - height (`int`, *optional*):
                     The height in pixels of the generated image.
-                    This is set to 1024 by default for the best results.
+                    This is set to 512 by default for the best results.
                 - width (`int`, *optional*):
                     The width in pixels of the generated image.
-                    This is set to 1024 by default for the best results.
+                    This is set to 512 by default for the best results.
                 - num_inference_steps (`int`, *optional*, defaults to 28):
                 - guidance_scale (`float`, *optional*, defaults to 3.5):
                     Guidance scale as defined in Classifier-Free Diffusion Guidance.