Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-05 20:40:44 +02:00
Code Issues Packages Projects Releases Wiki Activity

always keep lora on gpu

Browse Source
This commit is contained in:
Vladimir Mandic
2024-09-22 13:13:57 -04:00
parent 74a70a8e03
commit fdb2ea8260
9 changed files with 21 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing_args.py
+1
View File
@@ -268,6 +268,7 @@ def set_pipeline_args(p, model, prompts: list, negative_prompts: list, prompts_2
if isinstance(v, list) and len(v) > 0 and (isinstance(v[0], torch.Tensor) or isinstance(v[0], np.ndarray)):
clean[k] = [x.shape for x in v]
shared.log.debug(f'Diffuser pipeline: {model.__class__.__name__} task={sd_models.get_diffusers_task(model)} batch={p.iteration + 1}/{p.n_iter}x{p.batch_size} set={clean}')
if p.hdr_clamp or p.hdr_maximize or p.hdr_brightness != 0 or p.hdr_color != 0 or p.hdr_sharpen != 0:
txt = 'HDR:'
txt += f' Brightness={p.hdr_brightness}' if p.hdr_brightness != 0 else ' Brightness off'