Work on additional image embedding methods. Finalized zipper resampler. It works amazing

2024-02-10 09:00:05 -07:00
parent a8481c1670
commit e074058faa
7 changed files with 261 additions and 47 deletions
--- a/toolkit/network_mixins.py
+++ b/toolkit/network_mixins.py
@@ -473,7 +473,7 @@ class ToolkitNetworkMixin:
            del load_sd[key]

        print(f"Missing keys: {to_delete}")
-        if len(to_delete) > 0 and self.is_v1:
+        if len(to_delete) > 0 and self.is_v1 and not force_weight_mapping and not (len(to_delete) == 1 and 'emb_params' in to_delete):
            print(" Attempting to load with forced keymap")
            return self.load_weights(file, force_weight_mapping=True)