import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.2-klein-4B", dtype=torch.bfloat16, device_map="cuda")
pipe.load_lora_weights("AnimeOverlord/flux2-klein-4b-mc-v2")
prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt).images[0]Flux.2 [Klein] DreamBooth LoRA - AnimeOverlord/flux2-klein-4b-mc-v2
Model description
These are AnimeOverlord/flux2-klein-4b-mc-v2 DreamBooth LoRA weights for black-forest-labs/FLUX.2-klein-4B.
The weights were trained using DreamBooth with the Flux2 diffusers trainer.
FP8 training? False
Trigger words
You should use None to trigger the image generation.
Download model
Download the *.safetensors LoRA in the Files & versions tab.
Use it with the 🧨 diffusers library
from diffusers import AutoPipelineForText2Image
import torch
pipeline = AutoPipelineForText2Image.from_pretrained("black-forest-labs/FLUX.2", torch_dtype=torch.bfloat16).to('cuda')
pipeline.load_lora_weights('AnimeOverlord/flux2-klein-4b-mc-v2', weight_name='pytorch_lora_weights.safetensors')
image = pipeline('None').images[0]
For more details, including weighting, merging and fusing LoRAs, check the documentation on loading LoRAs in diffusers
License
Please adhere to the licensing terms as described here.
Intended uses & limitations
How to use
# TODO: add an example code snippet for running this diffusion pipeline
Limitations and bias
[TODO: provide examples of latent issues and potential remediations]
Training details
[TODO: describe the data used to train the model]
- Downloads last month
- 68
Model tree for AnimeOverlord/flux2-klein-4b-mc-v2
Base model
black-forest-labs/FLUX.2-klein-4B