Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Small QoL imporvements

Browse Source
This commit is contained in:
Benjamin Warner
2022-08-23 12:49:17 -05:00
parent a21156e3e3
commit de1cea92ce
3 changed files with 254 additions and 116 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
scripts/dream.py
View File

@ -57,7 +57,8 @@ def main():
weights=weights,
full_precision=opt.full_precision,
config=config,
latent_diffusion_weights=opt.laion400m # this is solely for recreating the prompt
latent_diffusion_weights=opt.laion400m, # this is solely for recreating the prompt
device=opt.device
)
# make sure the output directory exists
@ -268,6 +269,11 @@ def create_argv_parser():
type=str,
default="outputs/img-samples",
help="directory in which to place generated images and a log of prompts and seeds")
parser.add_argument('--device',
'-d',
type=str,
default="cuda",
help="device to run stable diffusion on. defaults to cuda `torch.cuda.current_device()` if avalible")
return parser