From 9ea29a769d5ad21e6e1cdff86672c9026a0a2302 Mon Sep 17 00:00:00 2001
From: Artiprocher <wangye87v5@hotmail.com>
Date: Mon, 14 Apr 2025 13:18:31 +0800
Subject: [PATCH] fix base model

---
 train_flux_reference.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/train_flux_reference.py b/train_flux_reference.py
index 8d15102..3d626af 100644
--- 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()