aboutsummaryrefslogtreecommitdiff
path: root/modules/esrgan_model.py
diff options
context:
space:
mode:
authorAUTOMATIC <16777216c@gmail.com>2022-10-02 21:21:10 +0300
committerAUTOMATIC <16777216c@gmail.com>2022-10-02 21:21:10 +0300
commit0a435f0988f729fad5ed5bf14ba033795396067b (patch)
tree2eed7a259e6af80261c5570cddef145ee84aa1e2 /modules/esrgan_model.py
parenta1cde7e6468f80584030525a1b07cbf0f4ee42eb (diff)
parent0b94fc5033b8fa5a2613bda21ca0adffbaf73313 (diff)
Merge remote-tracking branch 'origin/master'
Diffstat (limited to 'modules/esrgan_model.py')
-rw-r--r--modules/esrgan_model.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/modules/esrgan_model.py b/modules/esrgan_model.py
index ea91abfe..4aed9283 100644
--- a/modules/esrgan_model.py
+++ b/modules/esrgan_model.py
@@ -73,8 +73,8 @@ def fix_model_layers(crt_model, pretrained_net):
class UpscalerESRGAN(Upscaler):
def __init__(self, dirname):
self.name = "ESRGAN"
- self.model_url = "https://drive.google.com/u/0/uc?id=1TPrz5QKd8DHHt1k8SRtm6tMiPjz_Qene&export=download"
- self.model_name = "ESRGAN 4x"
+ self.model_url = "https://github.com/cszn/KAIR/releases/download/v1.0/ESRGAN.pth"
+ self.model_name = "ESRGAN_4x"
self.scalers = []
self.user_path = dirname
self.model_path = os.path.join(models_path, self.name)