import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", dtype=torch.bfloat16, device_map="cuda")
pipe.load_lora_weights("Reza2kn/XRAY")
prompt = "chest XRAY image of 50 year old male with bilateral secondary PTB with right upper atelectasis, right pleural adhesions, left compensatory emphysema"
image = pipe(prompt).images[0]XRAY
- Prompt
- chest XRAY image of 50 year old male with bilateral secondary PTB with right upper atelectasis, right pleural adhesions, left compensatory emphysema
Model description
Tryin' somethin' here. v1
Trigger words
You should use XRAY to trigger the image generation.
Download model
Weights for this model are available in Safetensors format.
Download them in the Files & versions tab.
- Downloads last month
- 14
Model tree for Reza2kn/XRAY
Base model
black-forest-labs/FLUX.1-dev