fix(api): include SD upscaling in diffusion prefixes
This commit is contained in:
parent
46d9fc0dd4
commit
293a1bb184
|
@ -26,6 +26,7 @@ from .utils import (
|
||||||
DEFAULT_OPSET,
|
DEFAULT_OPSET,
|
||||||
ConversionContext,
|
ConversionContext,
|
||||||
download_progress,
|
download_progress,
|
||||||
|
fix_diffusion_name,
|
||||||
remove_prefix,
|
remove_prefix,
|
||||||
source_format,
|
source_format,
|
||||||
tuple_to_correction,
|
tuple_to_correction,
|
||||||
|
@ -265,20 +266,6 @@ def fetch_model(
|
||||||
return source, False
|
return source, False
|
||||||
|
|
||||||
|
|
||||||
DIFFUSION_PREFIX = ["diffusion-", "diffusion/", "diffusion\\", "stable-diffusion-"]
|
|
||||||
|
|
||||||
|
|
||||||
def fix_diffusion_name(name: str):
|
|
||||||
if not any([name.startswith(prefix) for prefix in DIFFUSION_PREFIX]):
|
|
||||||
logger.warning(
|
|
||||||
"diffusion models must have names starting with diffusion- to be recognized by the server: %s does not match",
|
|
||||||
name,
|
|
||||||
)
|
|
||||||
return f"diffusion-{name}"
|
|
||||||
|
|
||||||
return name
|
|
||||||
|
|
||||||
|
|
||||||
def convert_models(conversion: ConversionContext, args, models: Models):
|
def convert_models(conversion: ConversionContext, args, models: Models):
|
||||||
model_errors = []
|
model_errors = []
|
||||||
|
|
||||||
|
|
|
@ -345,3 +345,17 @@ def onnx_export(
|
||||||
all_tensors_to_one_file=True,
|
all_tensors_to_one_file=True,
|
||||||
location=ONNX_WEIGHTS,
|
location=ONNX_WEIGHTS,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
DIFFUSION_PREFIX = ["diffusion-", "diffusion/", "diffusion\\", "stable-diffusion-", "upscaling-"]
|
||||||
|
|
||||||
|
|
||||||
|
def fix_diffusion_name(name: str):
|
||||||
|
if not any([name.startswith(prefix) for prefix in DIFFUSION_PREFIX]):
|
||||||
|
logger.warning(
|
||||||
|
"diffusion models must have names starting with diffusion- to be recognized by the server: %s does not match",
|
||||||
|
name,
|
||||||
|
)
|
||||||
|
return f"diffusion-{name}"
|
||||||
|
|
||||||
|
return name
|
||||||
|
|
Loading…
Reference in New Issue