mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
de7abce464
* add an argument that lets user specify folders to scan for weights
This PR adds a `--weight_folders` argument to invoke.py. Using
argparse, it adds a "weight_folders" attribute to the Args object, and
can be used like this:
```
'''test.py'''
from ldm.invoke.args import Args
args = Args().parse_args()
for folder in args.weight_folders:
print(folder)
```
Example output:
```
python test.py --weight_folders /tmp/weights /home/fred/invokeai/weights "./my folder with spaces/weight files"
/tmp/weights
/home/fred/invokeai/weights
./my folder with spaces/weight files
```
* change --weight_folders to --weight_dirs
|
||
|---|---|---|
| .. | ||
| generator | ||
| restoration | ||
| __init__.py | ||
| args.py | ||
| CLI.py | ||
| concepts_lib.py | ||
| conditioning.py | ||
| devices.py | ||
| globals.py | ||
| image_util.py | ||
| log.py | ||
| model_cache.py | ||
| pngwriter.py | ||
| prompt_parser.py | ||
| readline.py | ||
| seamless.py | ||
| server_legacy.py | ||
| server.py | ||
| txt2mask.py | ||