mirror of
https://github.com/vladmandic/automatic
synced 2026-09-18 16:54:33 +02:00
@@ -5,7 +5,7 @@ from typing import Union
|
||||
from diffusers import StableDiffusionPipeline, StableDiffusionXLPipeline, FluxPipeline, StableDiffusion3Pipeline, ControlNetModel
|
||||
from modules.control.units import detect
|
||||
from modules.shared import log, opts, cmd_opts, state, listdir
|
||||
from modules import errors, sd_models, devices
|
||||
from modules import errors, sd_models, devices, model_quant
|
||||
from modules.processing import StableDiffusionProcessingControl
|
||||
|
||||
|
||||
@@ -163,7 +163,7 @@ def find_models():
|
||||
find_models()
|
||||
|
||||
|
||||
def api_list_models(model_type: str | None = None):
|
||||
def api_list_models(model_type: str = None):
|
||||
import modules.shared
|
||||
model_type = model_type or modules.shared.sd_model_type
|
||||
model_list = []
|
||||
@@ -215,7 +215,7 @@ def list_models(refresh=False):
|
||||
|
||||
|
||||
class ControlNet():
|
||||
def __init__(self, model_id: str | None = None, device = None, dtype = None, load_config = None):
|
||||
def __init__(self, model_id: str = None, device = None, dtype = None, load_config = None):
|
||||
self.model: ControlNetModel = None
|
||||
self.model_id: str = model_id
|
||||
self.device = device
|
||||
@@ -311,7 +311,7 @@ class ControlNet():
|
||||
self.load_config['original_config_file '] = config_path
|
||||
self.model = cls.from_single_file(model_path, config=config, **self.load_config)
|
||||
|
||||
def load(self, model_id: str | None = None, force: bool = False) -> str:
|
||||
def load(self, model_id: str = None, force: bool = False) -> str:
|
||||
with load_lock:
|
||||
try:
|
||||
t0 = time.time()
|
||||
|
||||
Reference in New Issue
Block a user