torch.no_grad(), cleanup

2022-06-28 12:16:44 -04:00
parent 1e2649148b
commit aef24ea157
7 changed files with 24 additions and 115 deletions
@@ -1,7 +1,6 @@
 #!/bin/bash

-_pip=$(command -v pip pip3)
-$_pip install -r requirements.txt
+pip install -r requirements.txt

 mkdir -p pretrained

@@ -10,7 +9,7 @@ git lfs install
 git clone https://huggingface.co/dalle-mini/vqgan_imagenet_f16_16384 ./pretrained/vqgan

 # download dalle-mini and dalle mega
-$_pip install wandb
+pip install wandb
 wandb login
 wandb artifact get --root=./pretrained/dalle_bart_mini dalle-mini/dalle-mini/mini-1:v0
 wandb artifact get --root=./pretrained/dalle_bart_mega dalle-mini/dalle-mini/mega-1-fp16:v14