README.md

---
license: creativeml-openrail-m
tags:
- stable-diffusion
- stable-diffusion-diffusers
- text-to-image
inference: true
---

# 🧵 WaxFashionStableDiffusion
A custom trained Stable Diffusion model for **generating African Wax Print-inspired fashion patterns**.This model is part of an ongoing research and creative exploration into increasing representation of African fashion in AI. This project leverages Stable Diffusion to generate vibrant African Wax Print textile patterns. Trained on a curated synthetic dataset of wax print images, the model captures key stylistic elements such as bold color contrasts, geometric motifs, and cultural symbolism. It enables designers, researchers, and creators to generate fashion patterns using simple text prompts.

### 📌 Model Overview
    **Model Type**: Fine-tuned Stable Diffusion  
     **Framework**: diffusers, PyTorch  
    **Resolution**: 512x512 images  
    **Use Case**: Textile pattern design, creative fashion generation, cultural pattern synthesis  

### 🎯 Use Cases
    Generate synthetic wax fabric designs  
    Aid in textile prototyping and visual merchandising  
    Explore culturally rich fashion patterns using AI  

### 📁 Dataset
    This model was trained using the **DreamBooth** method on a synthetic dataset of African Wax Print patterns.  
    **Name**: AfricanWaxPatterns_5KDataset  
    **Type**: Image dataset   
    **Size**: ~5,000 curated samples  
    **Dataset Link**: paceailab/AfricanWaxPatterns_5KDataset  

### 🛠️ How to Use
    🔧 In Python (Colab or local):  
  
  ```py
  from diffusers import StableDiffusionPipeline
  import torch
  
  model_id = "paceailab/WaxFashionStableDiffusion"
  
  pipe = StableDiffusionPipeline.from_pretrained(
      model_id,
      torch_dtype=torch.float16
  ).to("cuda")
  
  prompt = "Beautiful African Wax Pattern with orange and black design"
  image = pipe(prompt, num_inference_steps=50).images[0]
  
  image.save("wax_pattern.png")
  ```

### 📸 Sample Outputs

    Prompt	Output
## “Beautiful African Wax Pattern with orange and black design”  
<img src="https://cdn-uploads.huggingface.co/production/uploads/662301442e1aa351ad6248c7/BgX9knfDdaA8iPIrPBEV5.png" width="300"/>

## “Bright African wax print with red and blue flowers”  
<img src="https://cdn-uploads.huggingface.co/production/uploads/662301442e1aa351ad6248c7/msB12u95N6AYG-IdNzHPU.png" width="300"/>

## “Traditional wax pattern with geometric black and gold”
<img src="https://cdn-uploads.huggingface.co/production/uploads/662301442e1aa351ad6248c7/3PaCG5k5yKhakjI3sZvb5.png" width="300"/>

### 🔍 Training Details
&nbsp;&nbsp;&nbsp;&nbsp;**Training Batch Size** : 1  
&nbsp;&nbsp;&nbsp;&nbsp;**Train Steps** : 400  
&nbsp;&nbsp;&nbsp;&nbsp;**Learning Rate** : 5e-6  
&nbsp;&nbsp;&nbsp;&nbsp;**Resolution** : 512  
&nbsp;&nbsp;&nbsp;&nbsp;**Training prompt** : "African Wax Patterns"  

### 🗂️ Files & Versioning
The repository includes essential components for loading and using the WaxFashionStableDiffusion model:  
&nbsp;&nbsp;&nbsp;&nbsp;**tokenizer/**: Contains files for converting text prompts into tokens that the model can understand and process.  
&nbsp;&nbsp;&nbsp;&nbsp;**feature_extractor/**: Responsible for preprocessing inputs, such as resizing images or normalizing data, before feeding them into the model.  
&nbsp;&nbsp;&nbsp;&nbsp;**scheduler/**: Manages the denoising process during inference, controlling how noise is added or removed to generate high-quality images.  
&nbsp;&nbsp;&nbsp;&nbsp;**README.md** – Contains project overview, usage instructions, and relevant details.  
&nbsp;&nbsp;&nbsp;&nbsp;**model_index.json** – Metadata file that defines how the model can be loaded using diffusers.  
&nbsp;&nbsp;&nbsp;&nbsp;**v1-5-pruned.ckpt / .safetensors** – Core model weights, available in both .ckpt (Checkpoint) and .safetensors (a safer, more efficient format) for model inference.  
&nbsp;&nbsp;&nbsp;&nbsp;**v1-inference.yaml** – Configuration file containing settings used during model inference.  
&nbsp;&nbsp;&nbsp;&nbsp;**text_encoder/, unet/, vae/, safety_checker/** – Directories containing trained model components, like the text encoder, UNet, VAE, and safety checker used to run the model.  
&nbsp;&nbsp;&nbsp;&nbsp;.**gitattributes** – Manages Git LFS (Large File Storage) for efficiently handling large model files.  

### 📚 Related Resources
&nbsp;&nbsp;&nbsp;&nbsp;🔗 [Stable Diffusion, base model](https://github.com/CompVis/stable-diffusion)
  
&nbsp;&nbsp;&nbsp;&nbsp;🔗 [Diffusers Library](https://github.com/huggingface/diffusers) 

### 💡 Citation
&nbsp;&nbsp;&nbsp;&nbsp;If you use this model, please cite or credit:
  
  ```py
  @misc{waxfashion2025,
    title={WaxFashionStableDiffusion},
    author={Pace AI Lab},
    year={2025},
    howpublished={\url{https://huggingface.co/paceailab/WaxFashionStableDiffusion}}
  }
  ```