diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-03-28 19:39:20 +0300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-03-28 19:39:20 +0300 |
| commit | f1db987e6a3d16e278ea44979b43e585d893237e (patch) | |
| tree | c130645b2c8de1c528b69e613c28aed860664c12 /modules/sd_models.py | |
| parent | e49c479819760af910eae4881619c7f119359a5f (diff) | |
| parent | 1f08600345298fac0bcb66cc215a81875a84d7b9 (diff) | |
Merge pull request #8958 from MrCheeze/variations-model
Add support for the unclip (Variations) models, unclip-h and unclip-l
Diffstat (limited to 'modules/sd_models.py')
| -rw-r--r-- | modules/sd_models.py | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/modules/sd_models.py b/modules/sd_models.py index 86218c08..c2b3405c 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -383,6 +383,14 @@ def repair_config(sd_config): elif shared.cmd_opts.upcast_sampling:
sd_config.model.params.unet_config.params.use_fp16 = True
+ if getattr(sd_config.model.params.first_stage_config.params.ddconfig, "attn_type", None) == "vanilla-xformers" and not shared.xformers_available:
+ sd_config.model.params.first_stage_config.params.ddconfig.attn_type = "vanilla"
+
+ # For UnCLIP-L, override the hardcoded karlo directory
+ if hasattr(sd_config.model.params, "noise_aug_config") and hasattr(sd_config.model.params.noise_aug_config.params, "clip_stats_path"):
+ karlo_path = os.path.join(paths.models_path, 'karlo')
+ sd_config.model.params.noise_aug_config.params.clip_stats_path = sd_config.model.params.noise_aug_config.params.clip_stats_path.replace("checkpoints/karlo_models", karlo_path)
+
sd1_clip_weight = 'cond_stage_model.transformer.text_model.embeddings.token_embedding.weight'
sd2_clip_weight = 'cond_stage_model.model.transformer.resblocks.0.attn.in_proj_weight'
|