remove todo

inbarhub · Apr 20, 2023 · 4a444f0 · 4a444f0
1 parent c5eb3ed
commit 4a444f0
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/main_run.py b/main_run.py
@@ -15,7 +15,7 @@
 
 if __name__ == "__main__":
     parser = argparse.ArgumentParser()
-    parser.add_argument("--device_num", type=int, default=2)
+    parser.add_argument("--device_num", type=int, default=0)
     parser.add_argument("--cfg_src", type=float, default=3.5)
     parser.add_argument("--cfg_tar", type=float, default=[15])
     parser.add_argument("--num_diffusion_steps", type=int, default=100)
@@ -146,7 +146,3 @@
                     save_full_path = os.path.join(save_path, image_name_png)
                     img.save(save_full_path)
 
-# TODO: Inbar:
-# (1) fix Rene's bug
-# (2) write more concisly the inversion_util
-# (3) Comments