Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix comment

Browse Source
This commit is contained in:
Lincoln Stein 2023-12-14 23:04:13 -05:00 committed by Kent Keirsey
parent ac3cf48d7f
commit 212dbaf9a2
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
invokeai/backend/model_management/util.py
View File

@ -59,7 +59,7 @@ def lora_token_vector_length(checkpoint: dict) -> int:
lora_token_vector_length = _get_shape_1(key, tensor, checkpoint) lora_token_vector_length = _get_shape_1(key, tensor, checkpoint)
elif key.startswith("lora_unet_") and ( elif key.startswith("lora_unet_") and (
"time_emb_proj.lora_down" in key "time_emb_proj.lora_down" in key
): # recognizes format at https://civitai.com/models/224641 work ): # recognizes format at https://civitai.com/models/224641
lora_token_vector_length = _get_shape_1(key, tensor, checkpoint) lora_token_vector_length = _get_shape_1(key, tensor, checkpoint)
elif key.startswith("lora_te") and "_self_attn_" in key: elif key.startswith("lora_te") and "_self_attn_" in key:
tmp_length = _get_shape_1(key, tensor, checkpoint) tmp_length = _get_shape_1(key, tensor, checkpoint)

2
invokeai/backend/model_manager/probe.py
View File

@ -401,7 +401,7 @@ class LoRACheckpointProbe(CheckpointProbeBase):
elif token_vector_length == 1024: elif token_vector_length == 1024:
return BaseModelType.StableDiffusion2 return BaseModelType.StableDiffusion2
elif token_vector_length == 1280: elif token_vector_length == 1280:
return BaseModelType.StableDiffusionXL # recognizes format at https://civitai.com/models/224641 work return BaseModelType.StableDiffusionXL # recognizes format at https://civitai.com/models/224641
elif token_vector_length == 2048: elif token_vector_length == 2048:
return BaseModelType.StableDiffusionXL return BaseModelType.StableDiffusionXL
else: else: