Update esrgan_model.py
Use alternate ESRGAN Model download path.
This commit is contained in:
parent
3ff0de2c59
commit
6365a41f59
|
@ -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)
|
||||
|
|
Loading…
Reference in New Issue