Skip to content

Commit 47383e5

Browse files
committed
fixed lora loading in high vram mode
1 parent 172f643 commit 47383e5

1 file changed

Lines changed: 9 additions & 7 deletions

File tree

modules/pipelines/worker.py

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1302,13 +1302,15 @@ def fmt_eta(sec):
13021302
),
13031303
)
13041304
)
1305-
move_model_to_device_with_memory_preservation(
1306-
studio_module.current_generator.transformer,
1307-
target_device=gpu,
1308-
preserved_memory_gb=settings.get("gpu_memory_preservation"),
1309-
)
1310-
if selected_loras:
1311-
studio_module.current_generator.move_lora_adapters_to_device(gpu)
1305+
1306+
move_model_to_device_with_memory_preservation(
1307+
studio_module.current_generator.transformer,
1308+
target_device=gpu,
1309+
preserved_memory_gb=settings.get("gpu_memory_preservation"),
1310+
)
1311+
1312+
if selected_loras:
1313+
studio_module.current_generator.move_lora_adapters_to_device(gpu)
13121314

13131315
from diffusers_helper.pipelines.k_diffusion_hunyuan import sample_hunyuan
13141316

0 commit comments

Comments
 (0)