From 69bddac035b1369c80b09d1b412a7f412d1a5f98 Mon Sep 17 00:00:00 2001 From: Mohan Zhang Date: Sun, 12 Jun 2022 08:03:33 -0400 Subject: [PATCH] model_512_link from huggingface is not stable or not working. Changed it to another mirror. Also included *.pyc in .gitignore --- .gitignore | 1 + Disco_Diffusion.ipynb | 2 +- disco.py | 2 +- 3 files changed, 3 insertions(+), 2 deletions(-) diff --git a/.gitignore b/.gitignore index 17f21111..f0dbd63c 100644 --- a/.gitignore +++ b/.gitignore @@ -10,6 +10,7 @@ settings.json __pycache__/ *.py[cod] *$py.class +*.pyc # C extensions *.so diff --git a/Disco_Diffusion.ipynb b/Disco_Diffusion.ipynb index 52d99235..36ca2a1b 100644 --- a/Disco_Diffusion.ipynb +++ b/Disco_Diffusion.ipynb @@ -1792,7 +1792,7 @@ " model_secondary_SHA = '983e3de6f95c88c81b2ca7ebb2c217933be1973b1ff058776b970f901584613a'\n", "\n", " model_256_link = 'https://openaipublic.blob.core.windows.net/diffusion/jul-2021/256x256_diffusion_uncond.pt'\n", - " model_512_link = 'https://v-diffusion.s3.us-west-2.amazonaws.com/512x512_diffusion_uncond_finetune_008100.pt'\n", + " model_512_link = 'https://the-eye.eu/public/AI/models/512x512_diffusion_unconditional_ImageNet/512x512_diffusion_uncond_finetune_008100.pt'\n", " model_secondary_link = 'https://v-diffusion.s3.us-west-2.amazonaws.com/secondary_model_imagenet_2.pth'\n", "\n", " model_256_link_fb = 'https://www.dropbox.com/s/9tqnqo930mpnpcn/256x256_diffusion_uncond.pt'\n", diff --git a/disco.py b/disco.py index 059d0167..59320e73 100644 --- a/disco.py +++ b/disco.py @@ -1742,7 +1742,7 @@ def download_models(diffusion_model,use_secondary_model,fallback=False): model_secondary_SHA = '983e3de6f95c88c81b2ca7ebb2c217933be1973b1ff058776b970f901584613a' model_256_link = 'https://openaipublic.blob.core.windows.net/diffusion/jul-2021/256x256_diffusion_uncond.pt' - model_512_link = 'https://v-diffusion.s3.us-west-2.amazonaws.com/512x512_diffusion_uncond_finetune_008100.pt' + model_512_link = 'https://the-eye.eu/public/AI/models/512x512_diffusion_unconditional_ImageNet/512x512_diffusion_uncond_finetune_008100.pt' model_secondary_link = 'https://v-diffusion.s3.us-west-2.amazonaws.com/secondary_model_imagenet_2.pth' model_256_link_fb = 'https://www.dropbox.com/s/9tqnqo930mpnpcn/256x256_diffusion_uncond.pt'