fix refiner

This commit is contained in:
Vladimir Mandic
2023-09-07 09:38:35 -04:00
parent 241b47fa72
commit 81ff379e2c
3 changed files with 3 additions and 3 deletions
+1 -1
View File
@@ -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...')
+1 -1
View File
@@ -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)
+1 -1
View File
@@ -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