--- license: creativeml-openrail-m library_name: diffusers tags: - stable-diffusion-xl - stable-diffusion-xl-diffusers - text-to-image - diffusers - diffusers-training - lora base_model: stabilityai/stable-diffusion-xl-base-1.0 inference: true --- # BRAIN with Alzheimer's MRI generation (LoRA text2image fine-tuning) These are LoRA adaption weights for stabilityai/stable-diffusion-xl-base-1.0. Original dataset: https://www.kaggle.com/datasets/tourist55/alzheimers-dataset-4-class-of-images. With this model, you can generate an MRI of the brain of 4 classes: 1) NonDemented 2) VeryMildDemented 3) MildDemented 4) ModerateDemeneted For generation use the following corresponding prompts (or any other modification) 1) MRI of NonDemented brain 2) MRI of VeryMildDemented brain 3) MRI of MildDemented brain 4) MRI of ModerateDemeneted brain ![img_0](./image_0.png) ![img_1](./image_1.png) ![img_2](./image_2.png) ![img_3](./image_3.png) ![img_4](./image_4.png) ![img_5](./image_5.png) LoRA for the text encoder was enabled: False. Special VAE used for training: madebyollin/sdxl-vae-fp16-fix. ## Intended uses & limitations #### How to use ```python # 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]