mirror of
https://github.com/kohya-ss/sd-scripts.git
synced 2026-04-18 01:30:02 +00:00
refactor: update imports to use safetensors_utils for memory-efficient operations
This commit is contained in:
@@ -14,8 +14,8 @@ from transformers import (
|
||||
from transformers.models.t5.modeling_t5 import T5Stack
|
||||
from accelerate import init_empty_weights
|
||||
|
||||
from library import model_util
|
||||
from library.utils import load_safetensors, setup_logging
|
||||
from library.safetensors_utils import load_safetensors
|
||||
from library.utils import setup_logging
|
||||
|
||||
setup_logging()
|
||||
import logging
|
||||
|
||||
Reference in New Issue
Block a user