| license: creativeml-openrail-m | |
| tags: | |
| - stable-diffusion | |
| - text-to-image | |
| - floorplan | |
| - architecture | |
| - design | |
| # Floorplan Generation Model v3 | |
| A fine-tuned Stable Diffusion model specialized in generating architectural floorplans. | |
| ## Usage | |
| ```python | |
| from diffusers import StableDiffusionPipeline | |
| import torch | |
| pipe = StableDiffusionPipeline.from_pretrained("bavlyyoussef/floorplan-model-v3") | |
| pipe = pipe.to("cuda" if torch.cuda.is_available() else "cpu") | |
| prompt = "modern two-bedroom apartment floorplan with living room and kitchen" | |
| image = pipe(prompt).images[0] | |
| image.save("floorplan.png") | |
| ``` | |
| ## Training Details | |
| - Base Model: Stable Diffusion v2.1 | |
| - Training Method: LoRA fine-tuning | |
| - Resolution: 512x512 | |
| - Specialization: Architectural floorplans | |