license: mit | |
tags: | |
- text-to-image | |
- diffusion | |
- wan2.1 | |
- fine-tuned | |
library_name: diffusers | |
pipeline_tag: text-to-image | |
# Fine-Tuned Text-to-Image Model | |
This model is a fine-tuned version based on Wan2.1-T2V-14B. | |
## Note on Model Weights | |
Due to disk space limitations, the full model weights may not be included in this repository. | |
The model was fine-tuned on a custom dataset for 10 steps with a learning rate of 0.0001. | |
## Loading | |
```python | |
from diffusers import DiffusionPipeline | |
model = DiffusionPipeline.from_pretrained("your-username/text-image") | |
``` | |
## Parameters | |
- Steps: 10 | |
- Learning rate: 0.0001 | |
- Optimizer: adamw_8bit | |