fix base model

2026-03-18 22:08:13 +00:00 · 2025-04-14 13:18:31 +08:00
parent 2a5355b7cb
commit 9ea29a769d
1 changed files with 1 additions and 0 deletions
--- a/train_flux_reference.py
+++ b/train_flux_reference.py
@@ -32,6 +32,7 @@ class LightningModel(LightningModelForT2ILoRA):
            
        self.pipe = FluxImagePipeline.from_model_manager(model_manager)
        self.pipe.reference_embedder = FluxReferenceEmbedder()
+        self.pipe.reference_embedder.init()
        
        if quantize is not None:
            self.pipe.dit.quantize()