preload script more robust for downloading gfpgan

2024-08-30 20:32:17 +00:00 · 2022-10-05 11:29:48 -04:00 · 2022-10-05 11:29:48 -04:00 · 488890e6bb
commit 488890e6bb
parent 3feda31d82
1 changed files with 1 additions and 0 deletions
--- a/scripts/preload_models.py
+++ b/scripts/preload_models.py
@ -72,6 +72,7 @@ if gfpgan:

        if not os.path.exists(model_dest):
            print('downloading gfpgan model file...')
+            os.makedirs(os.path.dirname(model_dest))
            urllib.request.urlretrieve(model_url,model_dest)
    except Exception:
        import traceback