update readme

README.md

@@ -15,29 +15,13 @@ tags:
 
 # Flux.1 Lite
 
-We are thrilled to announce the alpha release of Flux.1 Lite, an 8B parameter transformer model distilled from the FLUX.1-dev model.
-
-Our goal? To distill FLUX.1-dev further until we achieve to reduce the parameters to just 24 GB, so it can run smoothly on most consumer-grade GPU cards, making high-quality AI models accessible to everyone.
+We are thrilled to announce the alpha release of Flux.1 Lite, an 8B parameter transformer model distilled from the FLUX.1-dev model. This version uses 7 GB less RAM and runs 23% faster while maintaining the same precision (bfloat16) as the original model.
 
 ![Flux.1 Lite vs FLUX.1-dev](./sample_images/models_comparison.png)
 
-## Motivation
-
-As stated by other members of the community like [Ostris](https://ostris.com/2024/09/07/skipping-flux-1-dev-blocks/), it seems that blocks of the Flux1.dev transformer have a different contribution to the final image generation. To explore this, we analyzed the Mean Squared Error (MSE) between the input and output of each block, revealing significant variability. 
-
-Our findings? Not all blocks contribute equally. The results are striking: skipping just one of the early MMDIT blocks can significantly impact model performance, whereas skipping the rest of the blocks do not have a significant impact over the final image quality.
-
-![Flux.1 Lite generated image](./sample_images/skip_blocks/generated_img.png)
-![MSE MMDIT](./sample_images/skip_blocks/mse_mmdit_img.png)
-![MSE DIT](./sample_images/skip_blocks/mse_dit_img.png)
-
-Furthermore, as displayed in the following image, only when you skip one of the first MMDIT blocks, the performance of the model severely impacts the model's performance.
-![Skip one MMDIT block](./sample_images/skip_blocks/skip_one_MMDIT_block.png)
-![Skip one DIT block](./sample_images/skip_blocks/skip_one_DIT_block.png)
-
-## Text-to-Image Usage
+## Text-to-Image
 
-Flux.1 Lite is ready to unleash your creativity! For the best results, we recommend using a `guidance_scale` of 3.5 and setting `n_steps` between 22 and 30.
+Flux.1 Lite is ready to unleash your creativity! For the best results, we strongly **recommend using a `guidance_scale` of 3.5 and setting `n_steps` between 22 and 30**.
 
 ```python
 import torch
@@ -66,27 +50,48 @@ with torch.inference_mode():
         generator=torch.Generator(device="cpu").manual_seed(seed),
         num_inference_steps=n_steps,
         guidance_scale=guidance_scale,
-        height=1024,s
+        height=1024,
         width=1024,
     ).images[0]
 image.save("output.png")
 ```
 
+## Motivation
+
+Inspired by [Ostris](https://ostris.com/2024/09/07/skipping-flux-1-dev-blocks/) findings, we analyzed the mean squared error (MSE) between the input and output of each block to quantify their contribution to the final result, revealing significant variability.
+
+As Ostris pointed out, not all blocks contribute equally. While skipping just one of the early MMDIT blocks can significantly impact model performance, skipping the rest of the blocks does not have a significant impact over the final image quality.
+
+![Flux.1 Lite generated image](./sample_images/skip_blocks/generated_img.png)
+![MSE MMDIT](./sample_images/skip_blocks/mse_mmdit_img.png)
+![MSE DIT](./sample_images/skip_blocks/mse_dit_img.png)
+
+Furthermore, as displayed in the following image, only when you skip one of the first MMDIT blocks, the performance of the model severely impacts the model's performance.
+
+![Skip one MMDIT block](./sample_images/skip_blocks/skip_one_MMDIT_block.png)
+![Skip one DIT block](./sample_images/skip_blocks/skip_one_DIT_block.png)
+
+## Future work
+
+Stay tuned! Our goal is to distill FLUX.1-dev further until it can run smoothly on 24 GB consumer-grade GPU cards, maintaining its original precision (bfloat16), and running even faster, making high-quality AI models accessible to everyone.
+
 ## ComfyUI
+
 We've also crafted a ComfyUI workflow to make using Flux.1 Lite even more seamless! Find it in `comfy/flux.1-lite_workflow.json`.
 ![ComfyUI workflow](./comfy/flux.1-lite_workflow.png)
 
-## Checkpoints
-* `flux.1-lite-8B-alpha.safetensors`: Transformer checkpoint, in Flux original format.
-* `transformers/`: Contains distilled 8B transformer model, in diffusers format.
+The safetensors checkpoint is available here: [flux.1-lite-8B-alpha.safetensors](flux.1-lite-8B-alpha.safetensors)
+
+## Try it out at Freepik!
 
-## 🤗 Hugging Face space: 
-Flux.1 Lite demo host on [🤗 flux.1-lite](https://huggingface.co/spaces/Freepik/flux.1-lite)
+Our [AI generator](https://www.freepik.com/pikaso/ai-image-generator) is now powered by Flux.1 Lite!
 
 ## 🔥 News 🔥
-* Oct.18, 2024. Alpha 8B checkpoint and comparison demo 🤗 (i.e. [Flux.1 Lite](https://huggingface.co/spaces/Freepik/flux.1-lite)) is publicly available on [HuggingFace Repo](https://huggingface.co/Freepik/flux.1-lite-8B-alpha).
+
+* Oct 23, 2024. Alpha 8B checkpoint is publicly available on [HuggingFace Repo](https://huggingface.co/Freepik/flux.1-lite-8B-alpha).
 
 ## Citation
+
 If you find our work helpful, please cite it!
 
 ```bibtex