this fixes the inconsistent use of self.device, sometimes a str and sometimes an obj

This commit is contained in:
Lincoln Stein
2022-09-01 10:16:05 -04:00
parent dc30adfbb4
commit 01e05a98de
2 changed files with 31 additions and 38 deletions

View File

@ -60,7 +60,6 @@ def main():
# this is solely for recreating the prompt
latent_diffusion_weights=opt.laion400m,
embedding_path=opt.embedding_path,
device=opt.device,
)
# make sure the output directory exists
@ -376,13 +375,6 @@ def create_argv_parser():
type=str,
help='Path to a pre-trained embedding manager checkpoint - can only be set on command line',
)
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',
)
parser.add_argument(
'--prompt_as_dir',
'-p',