mirror of
https://github.com/willmiao/ComfyUI-Lora-Manager.git
synced 2026-03-26 07:35:44 -03:00
feat(unet_loader): move torch import inside methods for lazy loading
- Delay torch import until needed in load_unet and load_unet_gguf methods - This improves module loading performance by avoiding unnecessary imports - Maintains functionality while reducing initial import overhead
This commit is contained in:
@@ -1,7 +1,6 @@
|
|||||||
import logging
|
import logging
|
||||||
import os
|
import os
|
||||||
from typing import List, Tuple
|
from typing import List, Tuple
|
||||||
import torch
|
|
||||||
import comfy.sd
|
import comfy.sd
|
||||||
from ..utils.utils import get_checkpoint_info_absolute, _format_model_name_for_comfyui
|
from ..utils.utils import get_checkpoint_info_absolute, _format_model_name_for_comfyui
|
||||||
|
|
||||||
@@ -101,6 +100,8 @@ class UNETLoaderLM:
|
|||||||
Returns:
|
Returns:
|
||||||
Tuple of (MODEL,)
|
Tuple of (MODEL,)
|
||||||
"""
|
"""
|
||||||
|
import torch
|
||||||
|
|
||||||
# Get absolute path from cache using ComfyUI-style name
|
# Get absolute path from cache using ComfyUI-style name
|
||||||
unet_path, metadata = get_checkpoint_info_absolute(unet_name)
|
unet_path, metadata = get_checkpoint_info_absolute(unet_name)
|
||||||
|
|
||||||
@@ -143,6 +144,7 @@ class UNETLoaderLM:
|
|||||||
Returns:
|
Returns:
|
||||||
Tuple of (MODEL,)
|
Tuple of (MODEL,)
|
||||||
"""
|
"""
|
||||||
|
import torch
|
||||||
from .gguf_import_helper import get_gguf_modules
|
from .gguf_import_helper import get_gguf_modules
|
||||||
|
|
||||||
# Get ComfyUI-GGUF modules using helper (handles various import scenarios)
|
# Get ComfyUI-GGUF modules using helper (handles various import scenarios)
|
||||||
|
|||||||
Reference in New Issue
Block a user