diff options
| author | Zac Liu <liuguang@baai.ac.cn> | 2022-12-06 09:16:15 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-12-06 09:16:15 +0800 |
| commit | 3ebf977a6e4f478ab918e44506974beee32da276 (patch) | |
| tree | f68456207e5cd78718ec1e9c588ecdc22d568d81 /extensions-builtin/LDSR/preload.py | |
| parent | 231fb72872191ffa8c446af1577c9003b3d19d4f (diff) | |
| parent | 44c46f0ed395967cd3830dd481a2db759fda5b3b (diff) | |
Merge branch 'AUTOMATIC1111:master' into master
Diffstat (limited to 'extensions-builtin/LDSR/preload.py')
| -rw-r--r-- | extensions-builtin/LDSR/preload.py | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/extensions-builtin/LDSR/preload.py b/extensions-builtin/LDSR/preload.py new file mode 100644 index 00000000..d746007c --- /dev/null +++ b/extensions-builtin/LDSR/preload.py @@ -0,0 +1,6 @@ +import os
+from modules import paths
+
+
+def preload(parser):
+ parser.add_argument("--ldsr-models-path", type=str, help="Path to directory with LDSR model file(s).", default=os.path.join(paths.models_path, 'LDSR'))
|