test

2024-01-26 20:22:49 +08:00 · 2024-01-26 20:22:49 +08:00 · 4e67fb8444
parent 50f631c768
commit 4e67fb8444
1 changed files with 1 additions and 3 deletions
--- 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