Keiser41
/

Example_Based_Manga_Colorization

Model card Files Files and versions

Keiser41 commited on Oct 28, 2023

Commit

84022c3

·

1 Parent(s): 8ed6625

Update pintar.py

Files changed (1) hide show

pintar.py +1 -1

pintar.py CHANGED Viewed

@@ -60,7 +60,7 @@ if __name__ == "__main__":
     device = "cuda"
     input_image_dir = args.input
     output_directory = args.output
-    model_checkpoint_path = args.checkpoint
     reference_image_path = args.reference
     imgsize = 256

     device = "cuda"
     input_image_dir = args.input
     output_directory = args.output
+    ckpt_path = args.checkpoint
     reference_image_path = args.reference
     imgsize = 256