How to use rntprch/FetalBrainPlanesLoRASDXL with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", dtype=torch.bfloat16, device_map="cuda") pipe.load_lora_weights("rntprch/FetalBrainPlanesLoRASDXL") prompt = "trans thalamic plane, fetal ultrasound scan, fetal brain, quality good, artifacts speckle noise, calvarium rim present, midline echo present, thalami present, csp present, lateral ventricle margins partial, third ventricle not clearly seen, posterior fossa not clearly seen, limitations posterior fossa obscured by shadowing" image = pipe(prompt).images[0]
The community tab is the place to discuss and collaborate with the HF community!
