Skip to content

S
Stable Diffusion Webui
Commit e8bbc344 authored by AUTOMATIC1111's avatar AUTOMATIC1111 Committed by GitHub
1 Browse files
Options

Merge pull request #8824 from zimkjh/fix-variable-typo 

fix variable typo
parents 9d2551d5 92e173d4
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
modules/sd_models.py
View file @ e8bbc344
...@@ -178,7 +178,7 @@ def select_checkpoint(): ...@@ -178,7 +178,7 @@ def select_checkpoint():
return checkpoint_info return checkpoint_info
chckpoint_dict_replacements = { checkpoint_dict_replacements = {
'cond_stage_model.transformer.embeddings.': 'cond_stage_model.transformer.text_model.embeddings.', 'cond_stage_model.transformer.embeddings.': 'cond_stage_model.transformer.text_model.embeddings.',
'cond_stage_model.transformer.encoder.': 'cond_stage_model.transformer.text_model.encoder.', 'cond_stage_model.transformer.encoder.': 'cond_stage_model.transformer.text_model.encoder.',
'cond_stage_model.transformer.final_layer_norm.': 'cond_stage_model.transformer.text_model.final_layer_norm.', 'cond_stage_model.transformer.final_layer_norm.': 'cond_stage_model.transformer.text_model.final_layer_norm.',
...@@ -186,7 +186,7 @@ chckpoint_dict_replacements = { ...@@ -186,7 +186,7 @@ chckpoint_dict_replacements = {
def transform_checkpoint_dict_key(k): def transform_checkpoint_dict_key(k):
for text, replacement in chckpoint_dict_replacements.items(): for text, replacement in checkpoint_dict_replacements.items():
if k.startswith(text): if k.startswith(text):
k = replacement + k[len(text):] k = replacement + k[len(text):]
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment