VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-08 14:34:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor: update imports to use safetensors_utils for memory-efficient operations

Browse Source
This commit is contained in:
Kohya S
2025-09-13 21:07:11 +09:00
parent d831c88832
commit 4e2a80a6ca
7 changed files with 20 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
hunyuan_image_minimal_inference.py
View File

@@ -22,6 +22,7 @@ from library import hunyuan_image_models, hunyuan_image_text_encoder, hunyuan_im
from library import hunyuan_image_vae
from library.hunyuan_image_vae import HunyuanVAE2D
from library.device_utils import clean_memory_on_device, synchronize_device
from library.safetensors_utils import mem_eff_save_file
from networks import lora_hunyuan_image
@@ -29,7 +30,7 @@ lycoris_available = find_spec("lycoris") is not None
if lycoris_available:
from lycoris.kohya import create_network_from_weights
from library.utils import mem_eff_save_file, setup_logging
from library.utils import setup_logging
setup_logging()
import logging