kpsss34/klein-4B-3BIT-SFW

Usage:

📂 infer_dir/
└── generate_quant_flux2.py
└── 📂 local_model/
    │   ├── pipeline_config.json
    │   └── scheduler
    │   └── text_encoder
    │   └── tokenizer
    │   └── transformer
        └── vae

import os
import torch
from generate_quant_flux2 import load_pipeline

pipe = load_pipeline("./local_model", "cuda")

prompt = ("A asian woman with red accent stripe features padded straps, brand logo, and multiple compartments, with a built-in LED light for low-light visibility.")

seed = 0
img = pipe(
    prompt=prompt,
    num_inference_steps=4,
    guidance_scale=1.0,
    generator=torch.Generator("cpu").manual_seed(seed),
).images[0]

os.makedirs("outputs", exist_ok=True)
path = f"img_s{seed}.png"
img.save(path)
print("saved", path)