diff --git a/train_network.py b/train_network.py
index d64d7a1b..8d102ae8 100644
--- a/train_network.py
+++ b/train_network.py
@@ -774,9 +774,7 @@ class NetworkTrainer:
                         else:
                             raise NotImplementedError("multipliers for each sample is not supported yet")
                         # print(f"set multiplier: {multipliers}")
-                        print(type(network))
-
-                        network.set_multiplier(multipliers)
+                        accelerator.unwrap_model(network).set_multiplier(multipliers)
 
                     with torch.set_grad_enabled(train_text_encoder), accelerator.autocast():
                         # Get the text embedding for conditioning