mirror of
https://github.com/vladmandic/automatic
synced 2026-09-18 16:54:33 +02:00
fix refiner
This commit is contained in:
@@ -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...')
|
||||
|
||||
@@ -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)
|
||||
|
||||
|
||||
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user