config.py

from dataclasses import dataclass


import os
import sys

import numpy as np

weights_path = os.path.join(sys.path[0], 'weights')


@dataclass
class PipelineConfig:
    """Fashion Controlnet Pipeline Config."""

    base_model_path: str = 'stabilityai/stable-diffusion-xl-base-1.0'

    vae_path: str = 'madebyollin/sdxl-vae-fp16-fix'

    controlnet_path: str = os.path.join(weights_path, 'controlnet')

    segmentation_model_path: str = os.path.join(weights_path, 'cloth_segm.pth')

    max_seed = np.iinfo(np.int32).max