From 6877b460c4c6b266986f3fc039598e122330843b Mon Sep 17 00:00:00 2001 From: Artiprocher Date: Fri, 2 Aug 2024 13:47:07 +0800 Subject: [PATCH] fix bugs --- diffsynth/controlnets/processors.py | 1 - examples/train/README.md | 6 ++++++ 2 files changed, 6 insertions(+), 1 deletion(-) diff --git a/diffsynth/controlnets/processors.py b/diffsynth/controlnets/processors.py index ce16b5d..1d23c73 100644 --- a/diffsynth/controlnets/processors.py +++ b/diffsynth/controlnets/processors.py @@ -47,6 +47,5 @@ class Annotator: detect_resolution = self.detect_resolution if self.detect_resolution is not None else min(width, height) image = self.processor(image, detect_resolution=detect_resolution, image_resolution=min(width, height), **kwargs) image = image.resize((width, height)) - image.save("/mnt/zhongjie/DiffSynth-Studio-kolors/DiffSynth-Studio/input.jpg") return image diff --git a/examples/train/README.md b/examples/train/README.md index bbc20b7..354dd90 100644 --- a/examples/train/README.md +++ b/examples/train/README.md @@ -13,6 +13,12 @@ Image Examples of fine-tuned LoRA. The prompt is "一只小狗蹦蹦跳跳，周 |Without LoRA|![image_without_lora](https://github.com/modelscope/DiffSynth-Studio/assets/35051019/9d79ed7a-e8cf-4d98-800a-f182809db318)|![image_without_lora](https://github.com/modelscope/DiffSynth-Studio/assets/35051019/ddb834a5-6366-412b-93dc-6d957230d66e)|![image_without_lora](https://github.com/Artiprocher/DiffSynth-Studio/assets/35051019/1aa21de5-a992-4b66-b14f-caa44e08876e)| |With LoRA|![image_with_lora](https://github.com/modelscope/DiffSynth-Studio/assets/35051019/02f62323-6ee5-4788-97a1-549732dbe4f0)|![image_with_lora](https://github.com/modelscope/DiffSynth-Studio/assets/35051019/8e7b2888-d874-4da4-a75b-11b6b214b9bf)|![image_with_lora](https://github.com/Artiprocher/DiffSynth-Studio/assets/35051019/83a0a41a-691f-4610-8e7b-d8e17c50a282)| +## Install additional packages + +``` +pip install peft lightning +``` + ## Prepare your dataset We provide an example dataset [here](https://modelscope.cn/datasets/buptwq/lora-stable-diffusion-finetune/files). You need to manage the training images as follows: