fix to work with Diffusers 0.17.0

2026-04-09 06:45:09 +00:00 · 2023-06-11 16:57:17 +09:00
parent 7f6b581ef8
commit 4e25c8f78e
4 changed files with 86 additions and 343 deletions
--- a/library/train_util.py
+++ b/library/train_util.py
@@ -3467,11 +3467,11 @@ def sample_images(
        unet=unet,
        tokenizer=tokenizer,
        scheduler=scheduler,
-        clip_skip=args.clip_skip,
        safety_checker=None,
        feature_extractor=None,
        requires_safety_checker=False,
    )
+    pipeline.clip_skip = args.clip_skip     # Pipelineのコンストラクタにckip_skipを追加できないので後から設定する
    pipeline.to(device)

    save_dir = args.output_dir + "/sample"