Skip to content

MoonRide303/Fooocus-MRE

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Fooocus-MRE

image

Fooocus-MRE is an image generating software (based on Gradio), an enhanced variant of the original Fooocus dedicated for a bit more advanced users.

Fooocus-MRE is a rethinking of Stable Diffusion and Midjourney’s designs:

  • Learned from Stable Diffusion - the software is offline, open source, and free.

  • Learned from Midjourney - it provides high quality output with default settings, allowing users to focus on the prompts and images.

  • Learned from SD web UI and ComfyUI - more advanced users would like to have some control over image generation process.

Fooocus has included and automated lots of inner optimizations and quality improvements. Users can forget all those difficult technical parameters, and just enjoy the interaction between human and computer to "explore new mediums of thought and expanding the imaginative powers of the human species" [1].

Fooocus has simplified the installation. Between pressing "download" and generating the first image, the number of needed mouse clicks is strictly limited to less than 3. Minimal GPU memory requirement is 4GB (Nvidia).

Fooocus also developed many "fooocus-only" features for advanced users to get perfect results. Click here to browse the advanced features.

[1] David Holz, 2019.

Download

Windows

You can directly download Fooocus with:

>>> Click here to download <<<

After you download the file, please uncompress it, and then run the "run.bat".

image

In the first time you launch the software, it will automatically download models:

  1. It will download sd_xl_base_1.0_0.9vae.safetensors from here as the file "Fooocus\models\checkpoints\sd_xl_base_1.0_0.9vae.safetensors".
  2. It will download sd_xl_refiner_1.0_0.9vae.safetensors from here as the file "Fooocus\models\checkpoints\sd_xl_refiner_1.0_0.9vae.safetensors".
  3. Note that if you use inpaint, at the first time you inpaint an image, it will download Fooocus's own inpaint control model from here as the file "Fooocus\models\inpaint\inpaint.fooocus.patch" (the size of this file is 1.28GB).

image

If you already have these files, you can copy them to the above locations to speed up installation.

Note that if you see "MetadataIncompleteBuffer", then your model files are corrupted. Please download models again.

Below is a test on a relatively low-end laptop with 16GB System RAM and 6GB VRAM (Nvidia 3060 laptop). The speed on this machine is about 1.35 seconds per iteration. Pretty impressive – nowadays laptops with 3060 are usually at very acceptable price.

image

Besides, recently many other software report that Nvidia driver above 532 is sometimes 10x slower than Nvidia driver 531. If your generation time is very long, consider download Nvidia Driver 531 Laptop or Nvidia Driver 531 Desktop.

Note that the minimal requirement is 4GB Nvidia GPU memory (4GB VRAM) and 8GB system memory (8GB RAM). This requires using Microsoft’s Virtual Swap technique, which is automatically enabled by your Windows installation in most cases, so you often do not need to do anything about it. However, if you are not sure, or if you manually turned it off (would anyone really do that?), or if you see any "RuntimeError: CPUAllocator", you can enable it here:

Click here to the see the image instruction.

image

And make sure that you have at least 40GB free space on each drive if you still see "RuntimeError: CPUAllocator" !

Please open an issue if you use similar devices but still cannot achieve acceptable performances.

Colab

(Last tested - 2023 Sep 13)

Colab Info
Open In Colab Fooocus Colab (Official Version)
Open In Colab Fooocus-MRE Colab (MoonRide Edition)

Thanks to camenduru!

Linux (Using Anaconda)

If you want to use Anaconda/Miniconda, you can

git clone https://github.com/MoonRide303/Fooocus-MRE.git
cd Fooocus-MRE
conda env create -f environment.yaml
conda activate fooocus
pip install pygit2==1.12.2

Then download the models: download sd_xl_base_1.0_0.9vae.safetensors from here as the file "Fooocus\models\checkpoints\sd_xl_base_1.0_0.9vae.safetensors", and download sd_xl_refiner_1.0_0.9vae.safetensors from here as the file "Fooocus\models\checkpoints\sd_xl_refiner_1.0_0.9vae.safetensors". Or let Fooocus automatically download the models using the launcher:

conda activate fooocus
python entry_with_update.py

Or if you want to open a remote port, use

conda activate fooocus
python entry_with_update.py --listen

Linux (Using Python Venv)

Your Linux needs to have Python 3.10 installed, and lets say your Python can be called with command python3 with your venv system working, you can

git clone https://github.com/MoonRide303/Fooocus-MRE.git
cd Fooocus-MRE
python3 -m venv fooocus_env
source fooocus_env/bin/activate
pip install pygit2==1.12.2

See the above sections for model downloads. You can launch the software with:

source fooocus_env/bin/activate
python entry_with_update.py

Or if you want to open a remote port, use

source fooocus_env/bin/activate
python entry_with_update.py --listen

Linux (Using native system Python)

If you know what you are doing, and your Linux already has Python 3.10 installed, and your Python can be called with command python3 (and Pip with pip3), you can

git clone https://github.com/MoonRide303/Fooocus-MRE.git
cd Fooocus-MRE
pip3 install pygit2==1.12.2

See the above sections for model downloads. You can launch the software with:

python3 entry_with_update.py

Or if you want to open a remote port, use

python3 entry_with_update.py --listen

Linux (AMD GPUs)

Installation is the same as Linux part. It has been tested for 6700XT. Works for both Pytorch 1.13 and Pytorch 2.

Mac/Windows(AMD GPUs)

Coming soon ...

List of "Hidden" Tricks

Below things are already inside the software, and users do not need to do anything about these.

  1. GPT2-based prompt expansion as a dynamic style "Fooocus V2". (similar to Midjourney's hidden pre-processsing and "raw" mode, or the LeonardoAI's Prompt Magic).
  2. Native refiner swap inside one single k-sampler. The advantage is that now the refiner model can reuse the base model's momentum (or ODE's history parameters) collected from k-sampling to achieve more coherent sampling. In Automatic1111's high-res fix and ComfyUI's node system, the base model and refiner use two independent k-samplers, which means the momentum is largely wasted, and the sampling continuity is broken. Fooocus uses its own advanced k-diffusion sampling that ensures seamless, native, and continuous swap in a refiner setup. (Update Aug 13: Actually I discussed this with Automatic1111 several days ago and it seems that the “native refiner swap inside one single k-sampler” is merged into the dev branch of webui. Great!)
  3. Negative ADM guidance. Because the highest resolution level of XL Base does not have cross attentions, the positive and negative signals for XL's highest resolution level cannot receive enough contrasts during the CFG sampling, causing the results look a bit plastic or overly smooth in certain cases. Fortunately, since the XL's highest resolution level is still conditioned on image aspect ratios (ADM), we can modify the adm on the positive/negative side to compensate for the lack of CFG contrast in the highest resolution level. (Update Aug 16, the IOS App Drawing Things will support Negative ADM Guidance. Great!)
  4. We implemented a carefully tuned variation of the Section 5.1 of "Improving Sample Quality of Diffusion Models Using Self-Attention Guidance". The weight is set to very low, but this is Fooocus's final guarantee to make sure that the XL will never yield overly smooth or plastic appearance (examples here). This can almostly eliminate all cases that XL still occasionally produce overly smooth results even with negative ADM guidance. (Update 2023 Aug 18, the Gaussian kernel of SAG is changed to an anisotropic kernel for better structure preservation and fewer artifacts.)
  5. We modified the style templates a bit and added the "cinematic-default".
  6. We tested the "sd_xl_offset_example-lora_1.0.safetensors" and it seems that when the lora weight is below 0.5, the results are always better than XL without lora.
  7. The parameters of samplers are carefully tuned.
  8. Because XL uses positional encoding for generation resolution, images generated by several fixed resolutions look a bit better than that from arbitrary resolutions (because the positional encoding is not very good at handling int numbers that are unseen during training). This suggests that the resolutions in UI may be hard coded for best results.
  9. Separated prompts for two different text encoders seem unnecessary. Separated prompts for base model and refiner may work but the effects are random, and we refrain from implement this.
  10. DPM family seems well-suited for XL, since XL sometimes generates overly smooth texture but DPM family sometimes generate overly dense detail in texture. Their joint effect looks neutral and appealing to human perception.
  11. A carefully designed system for balancing multiple styles as well as prompt expansion.
  12. Using automatic1111's method to normalize prompt emphasizing. This significantly improve results when users directly copy prompts from civitai.
  13. The joint swap system of refiner now also support img2img and upscale in a seamless way.

Advanced Features

Click here to browse the advanced features.

MoonRide Edition Features

  1. Support for Image-2-Image mode.
  2. Support for Control-LoRA: Canny Edge (guiding diffusion using edge detection on input, see Canny Edge description from SAI).
  3. Support for Control-LoRA: Depth (guiding diffusion using depth information from input, see Depth description from SAI).
  4. Support for Control-LoRA: Revision (prompting with images, see Revision description from SAI).
  5. Adjustable text prompt strengths (useful in Revision mode).
  6. Support for embeddings (use "embedding:embedding_name" syntax, ComfyUI style).
  7. Customizable sampling parameters (sampler, scheduler, steps, base / refiner switch point, CFG, CLIP Skip).
  8. Displaying full metadata for generated images in the UI.
  9. Support for JPEG format.
  10. Ability to save full metadata for generated images (as JSON or embedded in image, disabled by default).
  11. Ability to load prompt information from JSON and image files (if saved with metadata).
  12. Ability to change default values of UI settings (loaded from settings.json file - use settings-example.json as a template).
  13. Ability to retain input files names (when using Image-2-Image mode).
  14. Ability to generate multiple images using same seed (useful in Image-2-Image mode).
  15. Ability to generate images forever (ported from SD web UI - right-click on Generate button to start or stop this mode).
  16. Official list of SDXL resolutions (as defined in SDXL paper).
  17. Compact resolution and style selection (thx to runew0lf for hints).
  18. Support for custom resolutions list (loaded from resolutions.json - use resolutions-example.json as a template).
  19. Support for custom resolutions - you can just type it now in Resolution field, like "1280x640".
  20. Support for upscaling via Image-2-Image (see example in Wiki).
  21. Support for custom styles (loaded from sdxl_styles folder on start).
  22. Support for playing audio when generation is finished (ported from SD web UI - use notification.ogg or notification.mp3).
  23. Starting generation via Ctrl-ENTER hotkey (ported from SD web UI).
  24. Support for loading models from subfolders (ported from RuinedFooocus).
  25. Support for authentication in --share mode (credentials loaded from auth.json - use auth-example.json as a template).
  26. Support for wildcards (ported from RuinedFooocus - put them in wildcards folder, then try prompts like __color__ sports car with different seeds).
  27. Support for FreeU.
  28. Limited support for non-SDXL models (no refiner, Control-LoRAs, Revision, inpainting, outpainting).
  29. Style Iterator (iterates over selected style(s) combined with remaining styles - S1, S1 + S2, S1 + S3, S1 + S4, and so on; for comparing styles pick no initial style, and use same seed for all images).

Thanks

The codebase starts from an odd mixture of Stable Diffusion web UI and ComfyUI. (And they both use GPL license.) MoonRide Edition is based on the original Fooocus. Big thanks to AUTOMATIC1111, comfyanonymous, and lllyasviel for providing those fantastic tools.

Thanks to Stability AI for researching and opening their Stable Diffusion model series, OpenAI for CLIP and mlfoundations for OpenCLIP, and LAION AI for data sets on which those models could learn.

Special thanks to twri and 3Diva for creating additional SDXL styles available in Fooocus.

Update Log

The log for original version is here, and for enhancements added in MRE here.

About

Focus on prompting and generating

Resources

License

Stars

Watchers

Forks

Sponsor this project

Languages

  • Python 96.6%
  • JavaScript 2.9%
  • Other 0.5%