diff --git a/config.py b/config.py index 203cf7d..3affb88 100644 --- a/config.py +++ b/config.py @@ -11,6 +11,8 @@ class RunConfig: sd_xl: bool = False # Whether to use Stable Diffusion v2.1 sd_2_1: bool = False + # Projection dimension of the SDXL text encoder used to reshape prompt embeddings + text_encoder_projection_dim: int = 1280 # Which token indices to alter with attend-and-excite token_indices: List[int] = None # Which random seeds to use when generating