From 81ff379e2cc34eec61a1957d92dc181c7e30fbda Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Thu, 7 Sep 2023 09:38:35 -0400 Subject: [PATCH] fix refiner --- launch.py | 2 +- modules/prompt_parser_diffusers.py | 2 +- modules/ui_extra_networks.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/launch.py b/launch.py index b6739bf5f..6afbd7458 100644 --- a/launch.py +++ b/launch.py @@ -216,7 +216,7 @@ if __name__ == "__main__": alive = False requests = 0 if round(time.time()) % 120 == 0: - installer.log.debug(f'Server alive={alive} Requests={requests} memory {get_memory_stats()} ') + installer.log.debug(f'Server alive={alive} requests={requests} memory {get_memory_stats()} ') if not alive: if instance.wants_restart: installer.log.info('Server restarting...') diff --git a/modules/prompt_parser_diffusers.py b/modules/prompt_parser_diffusers.py index 905169ca8..6a8b1bef0 100644 --- a/modules/prompt_parser_diffusers.py +++ b/modules/prompt_parser_diffusers.py @@ -168,7 +168,7 @@ def compel_encode_prompt( return prompt_embed, positive_pooled, negative_embed, negative_pooled elif 'XL' in pipeline.__class__.__name__ and is_refiner: - compel_te2 = Compel(tokenizer=pipeline.tokenizer_2, text_encoder=pipeline.text_encoder_2, returned_embeddings_type=embedding_type, requires_pooled=True, device=shared.device, textual_inversion_manager=textual_inversion_manager) + compel_te2 = Compel(tokenizer=pipeline.tokenizer_2, text_encoder=pipeline.text_encoder_2, returned_embeddings_type=embedding_type, requires_pooled=True, device=shared.device) positive, positive_pooled = compel_te2(prompt) negative, negative_pooled = compel_te2(negative_prompt) diff --git a/modules/ui_extra_networks.py b/modules/ui_extra_networks.py index c3a6e2ad9..17cf429b9 100644 --- a/modules/ui_extra_networks.py +++ b/modules/ui_extra_networks.py @@ -368,9 +368,9 @@ def create_ui(container, button, tabname, skip_indexing = False): res = [] for page in extra_pages: if title == '' or title == page.title or len(page.html) == 0: - shared.log.debug(f"Refreshing Extra networks: page={page.title} tab={ui.tabname}") page.refresh() page.create_page(ui.tabname) + shared.log.debug(f"Refreshing Extra networks: page={page.title} items={len(page.items)} tab={ui.tabname}") res.append(page.html) ui.search.update(value = ui.search.value) return res