Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 17:24:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

add missing js files

Browse Source 
Co-authored-by: Copilot <copilot@github.com>
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2026-05-02 10:56:52 +02:00
parent 823058b15a
commit 57f5bf5bd5
5 changed files with 17 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/prompt_enhance.py
+3 -3
View File
@@ -369,7 +369,7 @@ class PromptEnhanceScript(scripts_manager.Script):
sd_models.move_model(self.llm, devices.cpu, force=True)
self.llm = None
self.tokenizer = None
devices.torch_gc(force=True, reason='prompt enhance model switch')
devices.torch_gc(force=True, reason='prompt-enhance:load')
log.debug(f'Prompt enhance: name="{self.model}" unload')
self.model = None
load_args = { 'pretrained_model_name_or_path': model_repo if not gguf_args else model_gguf }
@@ -439,7 +439,7 @@ class PromptEnhanceScript(scripts_manager.Script):
self.model = None
self.llm = None
self.tokenizer = None
devices.torch_gc(force=True, reason='prompt enhance unload')
devices.torch_gc(force=True, reason='prompt-enhance:unload')
log.debug(f'Prompt enhance: model="{model_name}" unloaded')
else:
log.debug('Prompt enhance: no model loaded')
@@ -801,7 +801,7 @@ class PromptEnhanceScript(scripts_manager.Script):
response = f'Error: {str(e)}'
finally:
offload_aux('prompt_enhance')
devices.torch_gc(force=True, reason='prompt enhance offload')
devices.torch_gc(force=False, reason='prompt-enhance')
t1 = time.time()
if isinstance(response, list):