[Bugfix][Diffusion] Fuse distilled LoRA weights before HSDP sharding (#6948)
Signed-off-by: samithuang <285365963@qq.com> Signed-off-by: Samit <285365963@qq.com>
This commit is contained in:
1 parent
ae70479619
commit
44d3ae100a
9 files changed
+311
-1
No files matched your search
@@ -565,6 +565,7 @@ class DiffusionWorker:
|
||||
if self.od_config.lora_scale > 1.0:
|
||||
logger.warning("lora_scale > 1.0 may not take any effect when using distilled LoRA backend.")
|
||||
pipeline.load_lora_weights(lora_path)
|
||||
pipeline.lora_is_fused = True
|
||||
else:
|
||||
logger.warning("Pipeline does not support loading distilled LoRA weights for now.")
|
||||
else:
|
||||
|
||||
Reference in new issue
Block a user