Keiser41 commited on
Commit
84022c3
1 Parent(s): 8ed6625

Update pintar.py

Browse files
Files changed (1) hide show
  1. pintar.py +1 -1
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
- model_checkpoint_path = args.checkpoint
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