Instructions to use dx8152/Flux2-Klein-9B-Consistency with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use dx8152/Flux2-Klein-9B-Consistency with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline from diffusers.utils import load_image # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.2-klein-9B", dtype=torch.bfloat16, device_map="cuda") pipe.load_lora_weights("dx8152/Flux2-Klein-9B-Consistency") prompt = "Turn this cat into a dog" input_image = load_image("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/cat.png") image = pipe(image=input_image, prompt=prompt).images[0] - Inference
- Notebooks
- Google Colab
- Kaggle
- Local Apps Settings
- Draw Things
Can it be used on Diffusers
#12
by jj11 - opened
I used it on diffuers
transformer = Flux2Transformer2DModel.from_pretrained(
self.model_dir, subfolder="transformer", torch_dtype=torch.bfloat16,
device_map={"": 0},
)
pipe = replica_cls.from_pretrained(
self.model_dir, transformer=transformer, text_encoder=None,
torch_dtype=torch.bfloat16,
)
pipe.vae.to("cuda:0")
pipe.load_lora_weights(self.model_dir, weight_name="Flux2-Klein-9B-consistency-V2.safetensors")
The person generated and the one I uploaded are completely different. Can you please explain why this is happening