Update pintar.py
Browse files
pintar.py
CHANGED
@@ -60,7 +60,7 @@ if __name__ == "__main__":
|
|
60 |
device = "cuda"
|
61 |
input_image_dir = args.input
|
62 |
output_directory = args.output
|
63 |
-
|
64 |
reference_image_path = args.reference
|
65 |
imgsize = 256
|
66 |
|
|
|
60 |
device = "cuda"
|
61 |
input_image_dir = args.input
|
62 |
output_directory = args.output
|
63 |
+
ckpt_path = args.checkpoint
|
64 |
reference_image_path = args.reference
|
65 |
imgsize = 256
|
66 |
|