sync

app.py

@@ -216,7 +216,8 @@ def predict(
 
 @click.command()
 @click.option("--ckpt", type=str, default="checkpoints/v1-5-pruned-emaonly-adaption-task-humanalign.ckpt")
-def main(ckpt="checkpoints/v1-5-pruned-emaonly-adaption-task-humanalign.ckpt"):
+@click.option("--auto-download", type=bool, default=True)
+def main(ckpt="checkpoints/v1-5-pruned-emaonly-adaption-task-humanalign.ckpt", auto_download=True):
     css = '''
     .container {max-width: 1150px;margin: auto;padding-top: 1.5rem}
     #image_upload{min-height:400px}
@@ -256,6 +257,9 @@ def main(ckpt="checkpoints/v1-5-pruned-emaonly-adaption-task-humanalign.ckpt"):
     }
     '''
 
+    if auto_download:
+        os.system("bash scripts/download_instructdiffusion.sh")
+
     config = OmegaConf.load("configs/instruct_diffusion.yaml")
 
     # ckpt = "checkpoints/v1-5-pruned-emaonly-adaption-task-humanalign.ckpt"
@@ -333,4 +337,4 @@ def main(ckpt="checkpoints/v1-5-pruned-emaonly-adaption-task-humanalign.ckpt"):
 
 
 if __name__ == "__main__":
-    main()
+    main()

scripts/download_instructdiffusion.sh

@@ -0,0 +1,23 @@
+mkdir checkpoints
+
+wget https://github.com/TiankaiHang/storage-2023/releases/download/0924/v1-5-pruned-emaonly-adaption-task-humanalign_aa
+wget https://github.com/TiankaiHang/storage-2023/releases/download/0924/v1-5-pruned-emaonly-adaption-task-humanalign_ab
+wget https://github.com/TiankaiHang/storage-2023/releases/download/0924/v1-5-pruned-emaonly-adaption-task-humanalign_ac
+wget https://github.com/TiankaiHang/storage-2023/releases/download/0924/v1-5-pruned-emaonly-adaption-task-humanalign_ad
+wget https://github.com/TiankaiHang/storage-2023/releases/download/0924/v1-5-pruned-emaonly-adaption-task-humanalign_ae
+wget https://github.com/TiankaiHang/storage-2023/releases/download/0924/v1-5-pruned-emaonly-adaption-task-humanalign_af
+
+cat v1-5-pruned-emaonly-adaption-task-humanalign_* > checkpoints/v1-5-pruned-emaonly-adaption-task-humanalign.ckpt
+
+rm v1-5-pruned-emaonly-adaption-task-humanalign_*
+
+wget https://github.com/TiankaiHang/storage-2023/releases/download/0924/v1-5-pruned-emaonly-adaption-task_aa
+wget https://github.com/TiankaiHang/storage-2023/releases/download/0924/v1-5-pruned-emaonly-adaption-task_ab
+wget https://github.com/TiankaiHang/storage-2023/releases/download/0924/v1-5-pruned-emaonly-adaption-task_ac
+wget https://github.com/TiankaiHang/storage-2023/releases/download/0924/v1-5-pruned-emaonly-adaption-task_ad
+wget https://github.com/TiankaiHang/storage-2023/releases/download/0924/v1-5-pruned-emaonly-adaption-task_ae
+wget https://github.com/TiankaiHang/storage-2023/releases/download/0924/v1-5-pruned-emaonly-adaption-task_af
+
+cat v1-5-pruned-emaonly-adaption-task_* > checkpoints/v1-5-pruned-emaonly-adaption-task.ckpt
+
+rm v1-5-pruned-emaonly-adaption-task_*