| 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 | |