MODEL_NAME:str="stabilityai/stable-diffusion-xl-base-1.0"
ADAPTER_NAME:str = "VikramSingh178/sdxl-lora-finetune-product-caption"
ADAPTER_NAME_2:str = "VikramSingh178/Products10k-SDXL-Lora"
VAE_NAME:str= "madebyollin/sdxl-vae-fp16-fix"
DATASET_NAME:str = "hahminlew/kream-product-blip-captions"
PROJECT_NAME:str = "Product Photography"
PRODUCTS_10k_DATASET:str = "VikramSingh178/Products-10k-BLIP-captions"
CAPTIONING_MODEL_NAME:str = "Salesforce/blip-image-captioning-base"
SEGMENTATION_MODEL_NAME:str = "facebook/sam-vit-large"
DETECTION_MODEL_NAME:str = "yolov8l"
ENABLE_COMPILE:bool = False
INPAINTING_MODEL_NAME:str = 'kandinsky-community/kandinsky-2-2-decoder-inpaint'




class Config:
    def __init__(self):
        self.pretrained_model_name_or_path = MODEL_NAME
        self.pretrained_vae_model_name_or_path = VAE_NAME
        self.revision = None
        self.variant = None
        self.dataset_name = PRODUCTS_10k_DATASET
        self.dataset_config_name = None
        self.train_data_dir = None
        self.image_column = 'image'
        self.caption_column = 'text'
        self.validation_prompt = None
        self.num_validation_images = 4
        self.validation_epochs = 1
        self.max_train_samples = 7
        self.output_dir = "output"
        self.cache_dir = None
        self.seed = 42
        self.resolution = 512
        self.center_crop = True
        self.random_flip = True
        self.train_text_encoder = False
        self.train_batch_size = 64
        self.num_train_epochs = 400
        self.max_train_steps = None
        self.checkpointing_steps = 500
        self.checkpoints_total_limit = None
        self.resume_from_checkpoint = None
        self.gradient_accumulation_steps = 1
        self.gradient_checkpointing = False
        self.learning_rate = 1e-4
        self.scale_lr = False
        self.lr_scheduler = "constant"
        self.lr_warmup_steps = 500
        self.snr_gamma = None
        self.allow_tf32 = True
        self.dataloader_num_workers = 0
        self.use_8bit_adam = True
        self.adam_beta1 = 0.9
        self.adam_beta2 = 0.999
        self.adam_weight_decay = 1e-2
        self.adam_epsilon = 1e-08
        self.max_grad_norm = 1.0
        self.push_to_hub = True
        self.hub_token = None
        self.prediction_type = None
        self.hub_model_id = None
        self.logging_dir = "logs"
        self.report_to = "wandb"
        self.mixed_precision = 'fp16'
        self.local_rank = -1
        self.enable_xformers_memory_efficient_attention = False
        self.noise_offset = 0
        self.rank = 4
        self.debug_loss = False