update docs

.gitattributes

@@ -33,3 +33,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+flux1-lite_sample.png filter=lfs diff=lfs merge=lfs -text
+*.png filter=lfs diff=lfs merge=lfs -text

README.md

@@ -11,13 +11,14 @@ tags:
 - text-to-image
 ---
 
+![Flux.1 Lite](./sample_images/flux1-lite-8B_sample.png)
+
 # Flux.1 Lite
+
 We want to announce the alpha version of our new distilled Flux.1 Lite model, an 8B parameter transformer model distilled from the original Flux1. dev model.
 
 Our goal is to further reduce FLUX.1-dev transformer parameters up to 24Gb to make it compatible with most of GPU cards.
 
-![](./flux1-lite-8B_tearser.jpg)
-
 
 ## 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).
@@ -55,13 +56,13 @@ pipe = FluxPipeline.from_pretrained(
 prompt = "Scene inspired by 2000 comedy animation, a glowing green alien whose fluorescent skin emits light, standing in a dark purple forest. The alien is holding a large sign that reads 'LITE 8B ALPHA' in bold letters. The forest around is shadowy, with tall, eerie trees and mist rolling in. The alien radiates a soft, supernatural glow, illuminating the surroundings, creating a stark contrast between light and darkness. Style of an old comic, with flat colors, halftone shading, and a slightly weathered, vintage texture."
 
 guidance_scale = 3.5  # Important to keep guidance_scale to 3.5
-n_steps = 24
-seed = 12
+n_steps = 28
+seed = 11
 
 with torch.inference_mode():
     image = pipe(
         prompt=prompt,
-        generator=torch.Generator(device="cuda").manual_seed(seed),
+        generator=torch.Generator(device="cpu").manual_seed(seed),
         num_inference_steps=n_steps,
         guidance_scale=guidance_scale,
         height=1024,