From d404d59952decaeb73b16358fcecd44fc40a05e7 Mon Sep 17 00:00:00 2001 From: aiXander Date: Tue, 2 Apr 2024 02:19:01 -0700 Subject: [PATCH] adjust default training res --- trainer/config.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/trainer/config.py b/trainer/config.py index 3dc441b..3c0d8ea 100644 --- a/trainer/config.py +++ b/trainer/config.py @@ -13,7 +13,7 @@ class TrainingConfig(BaseModel): sd_model_version: Literal["sdxl", "sd15"] pretrained_model: dict = None seed: Union[int, None] = None - resolution: int = 768 + resolution: int = 512 validation_img_size: List[int] = [1024, 1024] train_batch_size: int = 4 num_train_epochs: int = 10000