apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

rename old converter

This commit is contained in:
Sean Sube 2023-12-24 06:24:45 -06:00
parent 5d31e85709
commit da6e813638
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
api/onnx_web/convert/__main__.py
View File

@ -18,7 +18,7 @@ from .client.huggingface import HuggingfaceClient
from .correction.gfpgan import convert_correction_gfpgan from .correction.gfpgan import convert_correction_gfpgan
from .diffusion.control import convert_diffusion_control from .diffusion.control import convert_diffusion_control
from .diffusion.diffusion import ( from .diffusion.diffusion import (
convert_diffusion_diffusers, convert_diffusion_diffusers_legacy,
convert_diffusion_diffusers_optimum, convert_diffusion_diffusers_optimum,
) )
from .diffusion.diffusion_xl import convert_diffusion_diffusers_xl from .diffusion.diffusion_xl import convert_diffusion_diffusers_xl
@ -59,11 +59,11 @@ Models = Dict[str, List[ModelDict]]
model_converters: Dict[str, Any] = { model_converters: Dict[str, Any] = {
"archive": convert_extract_archive, "archive": convert_extract_archive,
"img2img": convert_diffusion_diffusers, "img2img": convert_diffusion_diffusers_optimum,
"img2img-sdxl": convert_diffusion_diffusers_xl, "img2img-sdxl": convert_diffusion_diffusers_xl,
"inpaint": convert_diffusion_diffusers, "inpaint": convert_diffusion_diffusers_legacy,
"txt2img": convert_diffusion_diffusers_optimum, "txt2img": convert_diffusion_diffusers_optimum,
"txt2img-legacy": convert_diffusion_diffusers, "txt2img-legacy": convert_diffusion_diffusers_legacy,
"txt2img-sdxl": convert_diffusion_diffusers_xl, "txt2img-sdxl": convert_diffusion_diffusers_xl,
} }

2
api/onnx_web/convert/diffusion/diffusion.py
View File

@ -333,7 +333,7 @@ def collate_cnet(cnet_path):
@torch.no_grad() @torch.no_grad()
def convert_diffusion_diffusers( def convert_diffusion_diffusers_legacy(
conversion: ConversionContext, conversion: ConversionContext,
model: Dict, model: Dict,
format: Optional[str], format: Optional[str],