There was an error while loading. Please reload this page.
1 parent 172f643 commit 47383e5Copy full SHA for 47383e5
1 file changed
modules/pipelines/worker.py
@@ -1302,13 +1302,15 @@ def fmt_eta(sec):
1302
),
1303
)
1304
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)
+
+ move_model_to_device_with_memory_preservation(
+ studio_module.current_generator.transformer,
+ target_device=gpu,
+ preserved_memory_gb=settings.get("gpu_memory_preservation"),
+ )
1312
+ if selected_loras:
1313
+ studio_module.current_generator.move_lora_adapters_to_device(gpu)
1314
1315
from diffusers_helper.pipelines.k_diffusion_hunyuan import sample_hunyuan
1316
0 commit comments