import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.2-dev", dtype=torch.bfloat16, device_map="cuda")
pipe.load_lora_weights("danildushenev/pneumothorax-lora-flux-2-dev")
prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt).images[0]danildushenev/pneumothorax-lora-flux-2-dev
FLUX.2-dev LoRA adapter.
Files
pytorch_lora_weights.safetensors(SHA256:18c8e1c49ae8dc872ad7dd501e7af2baae11646edfe02b0ee3a092150cdbc5f0)
Usage (Diffusers)
import torch
from diffusers import Flux2Pipeline
pipe = Flux2Pipeline.from_pretrained("black-forest-labs/FLUX.2-dev", torch_dtype=torch.bfloat16).to("cuda")
pipe.load_lora_weights("danildushenev/pneumothorax-lora-flux-2-dev", weight_name="pytorch_lora_weights.safetensors")
image = pipe("your prompt", num_inference_steps=24, guidance_scale=2.5).images[0]
image.save("result.png")
Usage (ComfyUI)
- Download
pytorch_lora_weights.safetensorsfrom this repo. - Put it into
ComfyUI/models/loras/. - In workflow use LoRA Loader and set this file.
Trigger text: a photo of Human with pneumothorax
Generated on 2026-03-11 19:44:14Z.
- Downloads last month
- 7
Model tree for danildushenev/pneumothorax-lora-flux-2-dev
Base model
black-forest-labs/FLUX.2-dev