File size: 779 Bytes
a76452e cae8c4a a76452e 81b458f a76452e ed30d34 a76452e ed30d34 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
---
library_name: diffusers
---
# Model Card for Model ID
<!-- Provide a quick summary of what the model is/does. -->
The model is able to generate 512x512 breast cancer pathology images based on generic text prompts, eg. "A histopathology image of breast cancer tissue."
It is the fusion of Stable Diffusion 1.4 and a LORA fine-tuned on 28218 pathology images from the BRCA dataset.
## Model Details
### Model Description
The model is trained on Stable Diffusion 1.4 with a UNet architecture and a pretrained VAE. It uses a batch size of 16 and lora rank of 32.
### How to use
```python
pipe = DiffusionPipeline.from_pretrained(
"RiddleHe/SD14_pathology_base", torch_dtype=torch.float16
)
pipe.to('cuda')
prompt = "A histopathology image of breast cancer tissue"
```
|