fix ambiguous naming of self.device

This commit is contained in:
Lincoln Stein
2022-09-01 14:18:17 -04:00
2 changed files with 30 additions and 37 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',