Hugging Face's logo

Spaces:
zideliu
/
styledrop
Sleeping

App Files Community
3
zideliu commited on
Commit
0b6e063
•
1 Parent(s): 7f087f4

Update

Browse files
Files changed (2) hide show
  1. app.py +13 -5
  2. taming/models/vqgan.py +1 -1
app.py CHANGED
@@ -97,11 +97,19 @@ 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
- 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):
 
97
  print("downloading cc3m-285000.ckpt")
98
  os.makedirs("assets/ckpts/cc3m-285000.ckpt",exist_ok=True)
99
 
100
+ os.system("wget https://huggingface.co/nzl-thu/MUSE/resolve/main/assets/ckpts/cc3m-285000.ckpt/lr_scheduler.pth -O assets/ckpts/cc3m-285000.ckpt/lr_scheduler.pth")
101
+ os.system("wget https://huggingface.co/nzl-thu/MUSE/resolve/main/assets/ckpts/cc3m-285000.ckpt/optimizer.pth -O assets/ckpts/cc3m-285000.ckpt/optimizer.pth")
102
+ os.system("wget https://huggingface.co/nzl-thu/MUSE/resolve/main/assets/ckpts/cc3m-285000.ckpt/nnet.pth -O assets/ckpts/cc3m-285000.ckpt/nnet.pth")
103
+ os.system("wget https://huggingface.co/nzl-thu/MUSE/resolve/main/assets/ckpts/cc3m-285000.ckpt/nnet_ema.pth -O assets/ckpts/cc3m-285000.ckpt/nnet_ema.pth")
104
+ os.system("wget https://huggingface.co/nzl-thu/MUSE/resolve/main/assets/ckpts/cc3m-285000.ckpt/step.pth -O assets/ckpts/cc3m-285000.ckpt/step.pth")
105
+ os.system("wget https://huggingface.co/zideliu/vqgan/resolve/main/vqgan_jax_strongaug.ckpt -O assets/vqgan_jax_strongaug.ckpt")
106
+ os.system("ls assets/ckpts/cc3m-285000.ckpt")
107
+
108
+ # 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")
109
+ # wget.download("https://huggingface.co/nzl-thu/MUSE/resolve/main/assets/ckpts/cc3m-285000.ckpt/optimizer.pth","assets/ckpts/cc3m-285000.ckpt/optimizer.pth")
110
+ # wget.download("https://huggingface.co/nzl-thu/MUSE/resolve/main/assets/ckpts/cc3m-285000.ckpt/nnet.pth","assets/ckpts/cc3m-285000.ckpt/nnet.pth")
111
+ # 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")
112
+ # wget.download("https://huggingface.co/nzl-thu/MUSE/resolve/main/assets/ckpts/cc3m-285000.ckpt/step.pth","assets/ckpts/cc3m-285000.ckpt/step.pth")
113
  # wget.download("https://huggingface.co/zideliu/vqgan/resolve/main/vqgan_jax_strongaug.ckpt","assets/vqgan_jax_strongaug.ckpt")
114
  # os.system("ls assets/ckpts/cc3m-285000.ckpt")
115
  def set_seed(seed: int):
taming/models/vqgan.py CHANGED
@@ -111,4 +111,4 @@ def get_model(config_file='vq-f16-jax.yaml'):
111
  lossconfig=config.params.lossconfig,
112
  n_embed=config.params.n_embed,
113
  embed_dim=config.params.embed_dim,
114
- ckpt_path='assets/vqgan/vqgan_jax_strongaug.ckpt')
 
111
  lossconfig=config.params.lossconfig,
112
  n_embed=config.params.n_embed,
113
  embed_dim=config.params.embed_dim,
114
+ ckpt_path='assets/vqgan_jax_strongaug.ckpt')