Rewrite convert.py to work properly with dev de-distill

Original logic was triggering SDXL specific code to reshape img_in weight, causing forge not to work with it.
Tagging @blepping since this might change how the SDXL code is evaluated, but based on a quick check it should work.
This commit is contained in:
City
2024-10-15 03:56:54 +02:00
parent d2aaeb0f13
commit d090a910ac
2 changed files with 63 additions and 37 deletions
+1 -1
View File
@@ -67,7 +67,7 @@ def gguf_sd_loader(path, handle_prefix="model.diffusion_model."):
else: # stable-diffusion.cpp
# import here to avoid changes to convert.py breaking regular models
from .tools.convert import detect_arch
arch_str = detect_arch(set(val[0] for val in tensors))
arch_str = detect_arch(set(val[0] for val in tensors)).arch
compat = "sd.cpp"
# main loading loop