Update app.py

app.py

@@ -17,6 +17,45 @@ from diffusers import FluxPipeline
 
 torch.backends.cuda.matmul.allow_tf32 = True
 
+
+
+base_model_id = "Freepik/flux.1-lite-8B-alpha"
+torch_dtype = torch.bfloat16
+device = "cpu"
+
+# Load the pipe
+model_id = "Freepik/flux.1-lite-8B-alpha"
+pipe = FluxPipeline.from_pretrained(
+    model_id, torch_dtype=torch_dtype
+).to(device)
+
+# Inference
+prompt = "A close-up image of a green alien with fluorescent skin in the middle of a dark purple forest"
+
+guidance_scale = 3.5  # Keep guidance_scale at 3.5
+n_steps = 28
+seed = 11
+
+with torch.inference_mode():
+    image = pipe(
+        prompt=prompt,
+        generator=torch.Generator(device="cpu").manual_seed(seed),
+        num_inference_steps=n_steps,
+        guidance_scale=guidance_scale,
+        height=1024,
+        width=1024,
+    ).images[0]
+image.save("output.png")
+
+
+
+
+
+
+
+
+
+
 class timer:
     def __init__(self, method_name="timed process"):
         self.method = method_name