ysharma HF staff commited on
Commit
74f3a0d
1 Parent(s): 0019d94

update inversion,py

Browse files
Files changed (1) hide show
  1. src/inversion.py +4 -0
src/inversion.py CHANGED
@@ -58,7 +58,11 @@ if __name__=="__main__":
58
  torch_dtype=torch_dtype
59
  )
60
  # save the inversion
 
 
61
  torch.save(x_inv[0], os.path.join(args.results_folder, f"inversion/{bname}.pt"))
62
  # save the prompt string
 
 
63
  with open(os.path.join(args.results_folder, f"prompt/{bname}.txt"), "w") as f:
64
  f.write(prompt_str)
 
58
  torch_dtype=torch_dtype
59
  )
60
  # save the inversion
61
+ print("Inside inversion >> save the inversion >>>")
62
+ print(os.path.join(args.results_folder, f"inversion/{bname}.pt"))
63
  torch.save(x_inv[0], os.path.join(args.results_folder, f"inversion/{bname}.pt"))
64
  # save the prompt string
65
+ print("Inside inversion >> save the prompt string >>>")
66
+ print(os.path.join(args.results_folder, f"prompt/{bname}.txt"))
67
  with open(os.path.join(args.results_folder, f"prompt/{bname}.txt"), "w") as f:
68
  f.write(prompt_str)