Convert wan lora weights on save to be something comfy can handle

2026-01-26 16:39:47 +00:00 · 2025-03-08 12:55:11 -07:00
parent 7e37918fbc
commit e6739f7eb2
8 changed files with 108 additions and 5 deletions
--- a/jobs/process/BaseSDTrainProcess.py
+++ b/jobs/process/BaseSDTrainProcess.py
@@ -1611,6 +1611,7 @@ class BaseSDTrainProcess(BaseTrainProcess):
                    network_type=self.network_config.type,
                    transformer_only=self.network_config.transformer_only,
                    is_transformer=self.sd.is_transformer,
+                    base_model=self.sd,
                    **network_kwargs
                )