|
@@ -1,7 +1,7 @@
|
|
import os
|
|
import os
|
|
|
|
|
|
from modules import modelloader, errors
|
|
from modules import modelloader, errors
|
|
-from modules.shared import cmd_opts, opts
|
|
|
|
|
|
+from modules.shared import cmd_opts, opts, hf_endpoint
|
|
from modules.upscaler import Upscaler, UpscalerData
|
|
from modules.upscaler import Upscaler, UpscalerData
|
|
from modules.upscaler_utils import upscale_with_model
|
|
from modules.upscaler_utils import upscale_with_model
|
|
|
|
|
|
@@ -71,21 +71,21 @@ def get_dat_models(scaler):
|
|
return [
|
|
return [
|
|
UpscalerData(
|
|
UpscalerData(
|
|
name="DAT x2",
|
|
name="DAT x2",
|
|
- path="https://huggingface.co/w-e-w/DAT/resolve/main/experiments/pretrained_models/DAT/DAT_x2.pth",
|
|
|
|
|
|
+ path=f"{hf_endpoint}/w-e-w/DAT/resolve/main/experiments/pretrained_models/DAT/DAT_x2.pth",
|
|
scale=2,
|
|
scale=2,
|
|
upscaler=scaler,
|
|
upscaler=scaler,
|
|
sha256='7760aa96e4ee77e29d4f89c3a4486200042e019461fdb8aa286f49aa00b89b51',
|
|
sha256='7760aa96e4ee77e29d4f89c3a4486200042e019461fdb8aa286f49aa00b89b51',
|
|
),
|
|
),
|
|
UpscalerData(
|
|
UpscalerData(
|
|
name="DAT x3",
|
|
name="DAT x3",
|
|
- path="https://huggingface.co/w-e-w/DAT/resolve/main/experiments/pretrained_models/DAT/DAT_x3.pth",
|
|
|
|
|
|
+ path=f"{hf_endpoint}/w-e-w/DAT/resolve/main/experiments/pretrained_models/DAT/DAT_x3.pth",
|
|
scale=3,
|
|
scale=3,
|
|
upscaler=scaler,
|
|
upscaler=scaler,
|
|
sha256='581973e02c06f90d4eb90acf743ec9604f56f3c2c6f9e1e2c2b38ded1f80d197',
|
|
sha256='581973e02c06f90d4eb90acf743ec9604f56f3c2c6f9e1e2c2b38ded1f80d197',
|
|
),
|
|
),
|
|
UpscalerData(
|
|
UpscalerData(
|
|
name="DAT x4",
|
|
name="DAT x4",
|
|
- path="https://huggingface.co/w-e-w/DAT/resolve/main/experiments/pretrained_models/DAT/DAT_x4.pth",
|
|
|
|
|
|
+ path=f"{hf_endpoint}/w-e-w/DAT/resolve/main/experiments/pretrained_models/DAT/DAT_x4.pth",
|
|
scale=4,
|
|
scale=4,
|
|
upscaler=scaler,
|
|
upscaler=scaler,
|
|
sha256='391a6ce69899dff5ea3214557e9d585608254579217169faf3d4c353caff049e',
|
|
sha256='391a6ce69899dff5ea3214557e9d585608254579217169faf3d4c353caff049e',
|