VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-08 22:35:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix to work with Diffusers 0.17.0

Browse Source
This commit is contained in:
ykume
2023-06-11 16:57:17 +09:00
parent 7f6b581ef8
commit 4e25c8f78e
4 changed files with 86 additions and 343 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
library/lpw_stable_diffusion.py
View File

@@ -464,7 +464,7 @@ class StableDiffusionLongPromptWeightingPipeline(StableDiffusionPipeline):
tokenizer: CLIPTokenizer,
unet: UNet2DConditionModel,
scheduler: SchedulerMixin,
clip_skip: int,
# clip_skip: int,
safety_checker: StableDiffusionSafetyChecker,
feature_extractor: CLIPFeatureExtractor,
requires_safety_checker: bool = True,
@@ -479,7 +479,7 @@ class StableDiffusionLongPromptWeightingPipeline(StableDiffusionPipeline):
feature_extractor=feature_extractor,
requires_safety_checker=requires_safety_checker,
)
self.clip_skip = clip_skip
# self.clip_skip = clip_skip
self.__init__additional__()
# else: