Mirrors
/
waifu-diffusion
mirror of https://github.com/harubaru/waifu-diffusion.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #56 from harubaru/extended-mode-fix 

Extended Mode Typo Fix and Seed Update
This commit is contained in:
Anthony Mercurio 2022-12-01 20:47:18 -07:00 committed by GitHub
parent 138cb7bbed c4b17e41f1
commit c709257bec
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
trainer/diffusers_trainer.py
View File

@ -570,7 +570,7 @@ class AspectDataset(torch.utils.data.Dataset):
input_ids = z input_ids = z
else: else:
for i, x in enumerate(input_ids): for i, x in enumerate(input_ids):
input_ids[i] = [self.tokenizer.bos_token_id, *x, *np.full((self.max_length - len(x) - 1), self.tokenizer.eos_token_id)] input_ids[i] = [self.tokenizer.bos_token_id, *x, *np.full((self.tokenizer.model_max_length - len(x) - 1), self.tokenizer.eos_token_id)]
if args.clip_penultimate: if args.clip_penultimate:
input_ids = self.text_encoder.text_model.final_layer_norm(self.text_encoder(torch.asarray(input_ids).to(self.device), output_hidden_states=True)['hidden_states'][-2]) input_ids = self.text_encoder.text_model.final_layer_norm(self.text_encoder(torch.asarray(input_ids).to(self.device), output_hidden_states=True)['hidden_states'][-2])
else: else:
@ -717,6 +717,8 @@ def main():
# Set seed # Set seed
torch.manual_seed(args.seed) torch.manual_seed(args.seed)
random.seed(args.seed)
np.random.seed(args.seed)
print('RANDOM SEED:', args.seed) print('RANDOM SEED:', args.seed)
if args.resume: if args.resume: