add inversion script

snap-research · Jun 13, 2024 · 81ffab6 · 81ffab6
1 parent 9770e62
commit 81ffab6
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/inversion/invert.py b/inversion/invert.py
@@ -1,5 +1,6 @@
 import sys
-sys.path.append('/nfs/amil/peft/examples/w2w/')
+import os 
+sys.path.append(os.path.abspath(os.path.join("", "..")))
 import torch
 import torchvision
 import warnings
@@ -83,4 +84,4 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
+    main()