Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-09-01 19:49:03 +02:00
Code Issues Packages Projects Releases Wiki Activity

precommit

Browse Source
This commit is contained in:
XDUWQ
2023-07-25 15:00:28 +08:00
parent ffbf77fcf2
commit 3412a074c5
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
examples/pytorch/stable_diffusion/custom/finetune_stable_diffusion_custom.py
View File

@@ -94,7 +94,6 @@ class StableDiffusionCustomArguments(TrainingArgs):
metadata={
'help': 'Path to json containing multiple concepts.',
})
training_args = StableDiffusionCustomArguments(
@@ -160,7 +159,7 @@ pipe = pipeline(
task=Tasks.text_to_image_synthesis,
model=training_args.model,
custom_dir=training_args.work_dir + '/output',
modifier_token='<new1>',
modifier_token='<new1>+<new2>',
model_revision=args.model_revision)
output = pipe({'text': args.instance_prompt})

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

@@ -7,6 +7,7 @@ import warnings
from pathlib import Path
from typing import Union
import json
import numpy as np
import torch
import torch.nn.functional as F
@@ -309,9 +310,9 @@ class CustomDiffusionTrainer(EpochBasedTrainer):
'class_data_dir': self.class_data_dir,
}]
else:
with open(args.concepts_list, "r") as f:
with open(self.concepts_list, 'r') as f:
self.concepts_list = json.load(f)
print("--------self.concepts_list: ", self.concepts_list)
print('--------self.concepts_list: ', self.concepts_list)
# Adding a modifier token which is optimized
self.modifier_token_id = []