Kayson commited on
Commit
29cd0de
1 Parent(s): 2adcb54
Files changed (2) hide show
  1. app.py +6 -2
  2. scripts/download_instructdiffusion.sh +23 -0
app.py CHANGED
@@ -216,7 +216,8 @@ def predict(
216
 
217
  @click.command()
218
  @click.option("--ckpt", type=str, default="checkpoints/v1-5-pruned-emaonly-adaption-task-humanalign.ckpt")
219
- def main(ckpt="checkpoints/v1-5-pruned-emaonly-adaption-task-humanalign.ckpt"):
 
220
  css = '''
221
  .container {max-width: 1150px;margin: auto;padding-top: 1.5rem}
222
  #image_upload{min-height:400px}
@@ -256,6 +257,9 @@ def main(ckpt="checkpoints/v1-5-pruned-emaonly-adaption-task-humanalign.ckpt"):
256
  }
257
  '''
258
 
 
 
 
259
  config = OmegaConf.load("configs/instruct_diffusion.yaml")
260
 
261
  # 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"):
333
 
334
 
335
  if __name__ == "__main__":
336
- main()
 
216
 
217
  @click.command()
218
  @click.option("--ckpt", type=str, default="checkpoints/v1-5-pruned-emaonly-adaption-task-humanalign.ckpt")
219
+ @click.option("--auto-download", type=bool, default=True)
220
+ def main(ckpt="checkpoints/v1-5-pruned-emaonly-adaption-task-humanalign.ckpt", auto_download=True):
221
  css = '''
222
  .container {max-width: 1150px;margin: auto;padding-top: 1.5rem}
223
  #image_upload{min-height:400px}
 
257
  }
258
  '''
259
 
260
+ if auto_download:
261
+ os.system("bash scripts/download_instructdiffusion.sh")
262
+
263
  config = OmegaConf.load("configs/instruct_diffusion.yaml")
264
 
265
  # ckpt = "checkpoints/v1-5-pruned-emaonly-adaption-task-humanalign.ckpt"
 
337
 
338
 
339
  if __name__ == "__main__":
340
+ main()
scripts/download_instructdiffusion.sh ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ mkdir checkpoints
2
+
3
+ wget https://github.com/TiankaiHang/storage-2023/releases/download/0924/v1-5-pruned-emaonly-adaption-task-humanalign_aa
4
+ wget https://github.com/TiankaiHang/storage-2023/releases/download/0924/v1-5-pruned-emaonly-adaption-task-humanalign_ab
5
+ wget https://github.com/TiankaiHang/storage-2023/releases/download/0924/v1-5-pruned-emaonly-adaption-task-humanalign_ac
6
+ wget https://github.com/TiankaiHang/storage-2023/releases/download/0924/v1-5-pruned-emaonly-adaption-task-humanalign_ad
7
+ wget https://github.com/TiankaiHang/storage-2023/releases/download/0924/v1-5-pruned-emaonly-adaption-task-humanalign_ae
8
+ wget https://github.com/TiankaiHang/storage-2023/releases/download/0924/v1-5-pruned-emaonly-adaption-task-humanalign_af
9
+
10
+ cat v1-5-pruned-emaonly-adaption-task-humanalign_* > checkpoints/v1-5-pruned-emaonly-adaption-task-humanalign.ckpt
11
+
12
+ rm v1-5-pruned-emaonly-adaption-task-humanalign_*
13
+
14
+ wget https://github.com/TiankaiHang/storage-2023/releases/download/0924/v1-5-pruned-emaonly-adaption-task_aa
15
+ wget https://github.com/TiankaiHang/storage-2023/releases/download/0924/v1-5-pruned-emaonly-adaption-task_ab
16
+ wget https://github.com/TiankaiHang/storage-2023/releases/download/0924/v1-5-pruned-emaonly-adaption-task_ac
17
+ wget https://github.com/TiankaiHang/storage-2023/releases/download/0924/v1-5-pruned-emaonly-adaption-task_ad
18
+ wget https://github.com/TiankaiHang/storage-2023/releases/download/0924/v1-5-pruned-emaonly-adaption-task_ae
19
+ wget https://github.com/TiankaiHang/storage-2023/releases/download/0924/v1-5-pruned-emaonly-adaption-task_af
20
+
21
+ cat v1-5-pruned-emaonly-adaption-task_* > checkpoints/v1-5-pruned-emaonly-adaption-task.ckpt
22
+
23
+ rm v1-5-pruned-emaonly-adaption-task_*