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)
|