Spaces:
Sleeping
Sleeping
zideliu
commited on
Commit
•
7f087f4
1
Parent(s):
b19bf5d
update
Browse files
app.py
CHANGED
|
@@ -97,11 +97,11 @@ print("GPU memory:",torch.cuda.memory_allocated(0))
|
|
| 97 |
print("downloading cc3m-285000.ckpt")
|
| 98 |
os.makedirs("assets/ckpts/cc3m-285000.ckpt",exist_ok=True)
|
| 99 |
|
| 100 |
-
|
| 101 |
-
|
| 102 |
-
|
| 103 |
-
|
| 104 |
-
|
| 105 |
# wget.download("https://huggingface.co/zideliu/vqgan/resolve/main/vqgan_jax_strongaug.ckpt","assets/vqgan_jax_strongaug.ckpt")
|
| 106 |
# os.system("ls assets/ckpts/cc3m-285000.ckpt")
|
| 107 |
def set_seed(seed: int):
|
|
|
|
| 97 |
print("downloading cc3m-285000.ckpt")
|
| 98 |
os.makedirs("assets/ckpts/cc3m-285000.ckpt",exist_ok=True)
|
| 99 |
|
| 100 |
+
wget.download("https://huggingface.co/nzl-thu/MUSE/resolve/main/assets/ckpts/cc3m-285000.ckpt/lr_scheduler.pth","assets/ckpts/cc3m-285000.ckpt/lr_scheduler.pth")
|
| 101 |
+
wget.download("https://huggingface.co/nzl-thu/MUSE/resolve/main/assets/ckpts/cc3m-285000.ckpt/optimizer.pth","assets/ckpts/cc3m-285000.ckpt/optimizer.pth")
|
| 102 |
+
wget.download("https://huggingface.co/nzl-thu/MUSE/resolve/main/assets/ckpts/cc3m-285000.ckpt/nnet.pth","assets/ckpts/cc3m-285000.ckpt/nnet.pth")
|
| 103 |
+
wget.download("https://huggingface.co/nzl-thu/MUSE/resolve/main/assets/ckpts/cc3m-285000.ckpt/nnet_ema.pth","assets/ckpts/cc3m-285000.ckpt/nnet_ema.pth")
|
| 104 |
+
wget.download("https://huggingface.co/nzl-thu/MUSE/resolve/main/assets/ckpts/cc3m-285000.ckpt/step.pth","assets/ckpts/cc3m-285000.ckpt/step.pth")
|
| 105 |
# wget.download("https://huggingface.co/zideliu/vqgan/resolve/main/vqgan_jax_strongaug.ckpt","assets/vqgan_jax_strongaug.ckpt")
|
| 106 |
# os.system("ls assets/ckpts/cc3m-285000.ckpt")
|
| 107 |
def set_seed(seed: int):
|