mirror of
https://github.com/snapotter-hq/SnapOtter.git
synced 2026-08-03 07:46:42 +02:00
fix(docker): rename duplicate download_codeformer_model function
The ONNX variant shadowed the .pth variant due to identical function names, so codeformer.pth was never downloaded. Renamed the ONNX function to download_codeformer_onnx_model so both run.
This commit is contained in:
@@ -253,7 +253,7 @@ def download_ddcolor_model():
|
|||||||
|
|
||||||
|
|
||||||
|
|
||||||
def download_codeformer_model():
|
def download_codeformer_onnx_model():
|
||||||
"""Download CodeFormer ONNX model for AI face restoration.
|
"""Download CodeFormer ONNX model for AI face restoration.
|
||||||
|
|
||||||
Uses the pre-converted ONNX model from HuggingFace (facefusion repo)
|
Uses the pre-converted ONNX model from HuggingFace (facefusion repo)
|
||||||
@@ -594,7 +594,7 @@ def main():
|
|||||||
download_gfpgan_model()
|
download_gfpgan_model()
|
||||||
download_codeformer_model()
|
download_codeformer_model()
|
||||||
download_ddcolor_model()
|
download_ddcolor_model()
|
||||||
download_codeformer_model()
|
download_codeformer_onnx_model()
|
||||||
download_paddleocr_models()
|
download_paddleocr_models()
|
||||||
download_paddleocr_vl_model()
|
download_paddleocr_vl_model()
|
||||||
download_scunet_model()
|
download_scunet_model()
|
||||||
|
|||||||
Reference in New Issue
Block a user