Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2025-12-25 04:29:22 +01:00
Code Issues Packages Projects Releases Wiki Activity

custom diffusion

Browse Source
This commit is contained in:
XDUWQ
2023-07-12 17:35:57 +08:00
parent c1aa120029
commit cfa363d433
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
modelscope/trainers/multi_modal/custom_diffusion/custom_diffusion_trainer.py
View File

@@ -585,9 +585,8 @@ class CustomDiffusionTrainer(EpochBasedTrainer):
index_grads_to_zero = torch.arange(
len(self.model.tokenizer)) != self.modifier_token_id[0]
for i in range(len(self.modifier_token_id[1:])):
index_grads_to_zero = index_grads_to_zero & (
torch.arange(len(self.model.tokenizer)) !=
self.modifier_token_id[i])
modifier_flag = torch.arange(len(self.model.tokenizer)) != self.modifier_token_id[i]
index_grads_to_zero = index_grads_to_zero & modifier_flag
grads_data = grads_text_encoder.data[
index_grads_to_zero, :].fill_(0)
grads_text_encoder.data[