Merge branch 'main' into fit-init-img

add a --fit option to limit the size of the initial image to the
maximum boundaries specified by width and height.
This commit is contained in:
Lincoln Stein
2022-09-01 14:09:46 -04:00
5 changed files with 90 additions and 69 deletions

View File

@ -200,7 +200,7 @@ def main():
config = OmegaConf.load(f"{opt.config}")
model = load_model_from_config(config, f"{opt.ckpt}")
device = choose_torch_device()
device = torch.device(choose_torch_device())
model = model.to(device)
if opt.plms: