Merge pull request #1084 from fireicewolf/devel

Fix network multiplier cause crashed while use multi-GPUs

Author: kohya-ss

train_network.py

@@ -774,7 +774,7 @@ def remove_model(old_ckpt_name):
                         else:
                             raise NotImplementedError("multipliers for each sample is not supported yet")
                         # print(f"set multiplier: {multipliers}")
-                        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