diff --git a/TODO.md b/TODO.md index c178e4c20..6f24e1a1a 100644 --- a/TODO.md +++ b/TODO.md @@ -42,6 +42,7 @@ Stuff to be added, in no particular order... - Style editor (use json format instead of csv) - Profile manager (for config.json and ui-config.json) - Multi-user support + - Add [SAG](https://huggingface.co/docs/diffusers/v0.19.3/en/api/pipelines/self_attention_guidance),(https://github.com/ashen-sensored/sd_webui_SAG) - Image phash and hdash using `imagehash` - Model merge using `git-rebasin` - Enable refiner-style workflow for `ldm` backend diff --git a/cli/train.py b/cli/train.py index 58d97fa15..76de6e632 100755 --- a/cli/train.py +++ b/cli/train.py @@ -375,7 +375,7 @@ def check_versions(): log.info('checking accelerate') error = False import accelerate - if accelerate.__version__ != '0.19.0': + if accelerate.__version__ != '0.20.3': log.error(f'invalid accelerate version: accelerate=0.19.0 found={accelerate.__version__}') error = True log.info('checking diffusers') @@ -384,7 +384,7 @@ def check_versions(): log.error(f'invalid diffusers version: diffusers=0.10.2 found={diffusers.__version__}') error = True if error: - log.info('> pip install accelerate==0.19.0 diffusers==0.10.2') + log.info('> pip install accelerate==0.20.3 diffusers==0.10.2') exit(1) diff --git a/modules/prompt_parser_diffusers.py b/modules/prompt_parser_diffusers.py index eaa542c35..8850fb67b 100644 --- a/modules/prompt_parser_diffusers.py +++ b/modules/prompt_parser_diffusers.py @@ -99,6 +99,7 @@ def compel_encode_prompt( text_encoder=pipeline.text_encoder, returned_embeddings_type=embedding_type, requires_pooled=False, + truncate_long_prompts=False, device=shared.device )