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