from pydantic import BaseModel

class Config(BaseModel):
    clip_model_name: str = "sentence-transformers/clip-ViT-B-32"
    outfit_model_name: str = "elalber2000/clip-fashion-garments2look"
    
    minimum_similarity: float = 0.10
    minimum_margin: float = 0.0
    max_outfit_clothes: int = 4
    local_emb_path: str = "/mnt/c/Users/ASUS/Downloads/clip-iqon-1hour-model/model"

    train_app_name: str = "garments-clip-contrastive"
    train_model_name: str = "sentence-transformers/clip-ViT-B-32"
    train_data_mnt: str = "/data"
    train_output_mnt: str = "/outputs"
    train_hf_cache_mnt: str = "/root/.cache/huggingface"
    train_gpu: str = "A10G"
    train_timeout: int = 70 * 60
    train_memory: int = 32768

cnf = Config()