apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): patch download fn in facexlib (#134)

This commit is contained in:
Sean Sube 2023-02-13 18:30:09 -06:00
parent 6d503ca00e
commit c6b2751cc1
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 14 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
api/onnx_web/server/hacks.py
View File

@ -6,6 +6,7 @@ from urllib.parse import urlparse
import basicsr.utils.download_util import basicsr.utils.download_util
import codeformer.facelib.utils.misc import codeformer.facelib.utils.misc
import facexlib.utils
from ..utils import ServerContext from ..utils import ServerContext
@ -99,19 +100,25 @@ def patch_cache_path(ctx: ServerContext, url: str, **kwargs) -> str:
return cache_path return cache_path
def apply_patch_codeformer(ctx: ServerContext):
logger.debug("Patching CodeFormer module...")
codeformer.facelib.utils.misc.download_pretrained_models = patch_not_impl
codeformer.facelib.utils.misc.load_file_from_url = partial(patch_cache_path, ctx)
def apply_patch_basicsr(ctx: ServerContext): def apply_patch_basicsr(ctx: ServerContext):
logger.debug("Patching BasicSR module...") logger.debug("Patching BasicSR module...")
basicsr.utils.download_util.download_file_from_google_drive = patch_not_impl basicsr.utils.download_util.download_file_from_google_drive = patch_not_impl
basicsr.utils.download_util.load_file_from_url = partial(patch_cache_path, ctx) basicsr.utils.download_util.load_file_from_url = partial(patch_cache_path, ctx)
def apply_patch_codeformer(ctx: ServerContext):
logger.debug("Patching CodeFormer module...")
codeformer.facelib.utils.misc.download_pretrained_models = patch_not_impl
codeformer.facelib.utils.misc.load_file_from_url = partial(patch_cache_path, ctx)
def apply_patch_facexlib(ctx: ServerContext):
logger.debug("Patching Facexlib module...")
facexlib.utils.load_file_from_url = partial(patch_cache_path, ctx)
def apply_patches(ctx: ServerContext): def apply_patches(ctx: ServerContext):
apply_patch_basicsr(ctx) apply_patch_basicsr(ctx)
apply_patch_codeformer(ctx) apply_patch_codeformer(ctx)
unload(["basicsr.utils.download_util", "codeformer.facelib.utils.misc"]) apply_patch_facexlib(ctx)
unload(["basicsr.utils.download_util", "codeformer.facelib.utils.misc", "facexlib.utils"])