Spaces:
Paused
Paused
update inversion,py
Browse files- 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)
|