Skip to content

Commit 7f1c30b

Browse files
committed
use shared.hf_endpoin
1 parent 91de919 commit 7f1c30b

File tree

3 files changed

+8
-8
lines changed

3 files changed

+8
-8
lines changed

modules/dat_model.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
import os
22

33
from modules import modelloader, errors
4-
from modules.shared import cmd_opts, opts
4+
from modules.shared import cmd_opts, opts, hf_endpoint
55
from modules.upscaler import Upscaler, UpscalerData
66
from modules.upscaler_utils import upscale_with_model
77

@@ -71,21 +71,21 @@ def get_dat_models(scaler):
7171
return [
7272
UpscalerData(
7373
name="DAT x2",
74-
path="https://huggingface.co/w-e-w/DAT/resolve/main/experiments/pretrained_models/DAT/DAT_x2.pth",
74+
path=f"{hf_endpoint}/w-e-w/DAT/resolve/main/experiments/pretrained_models/DAT/DAT_x2.pth",
7575
scale=2,
7676
upscaler=scaler,
7777
sha256='7760aa96e4ee77e29d4f89c3a4486200042e019461fdb8aa286f49aa00b89b51',
7878
),
7979
UpscalerData(
8080
name="DAT x3",
81-
path="https://huggingface.co/w-e-w/DAT/resolve/main/experiments/pretrained_models/DAT/DAT_x3.pth",
81+
path=f"{hf_endpoint}/w-e-w/DAT/resolve/main/experiments/pretrained_models/DAT/DAT_x3.pth",
8282
scale=3,
8383
upscaler=scaler,
8484
sha256='581973e02c06f90d4eb90acf743ec9604f56f3c2c6f9e1e2c2b38ded1f80d197',
8585
),
8686
UpscalerData(
8787
name="DAT x4",
88-
path="https://huggingface.co/w-e-w/DAT/resolve/main/experiments/pretrained_models/DAT/DAT_x4.pth",
88+
path=f"{hf_endpoint}/w-e-w/DAT/resolve/main/experiments/pretrained_models/DAT/DAT_x4.pth",
8989
scale=4,
9090
upscaler=scaler,
9191
sha256='391a6ce69899dff5ea3214557e9d585608254579217169faf3d4c353caff049e',

modules/models/sd3/sd3_cond.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ def __getitem__(self, key):
2424
return self.file.get_tensor(key)
2525

2626

27-
CLIPL_URL = "https://huggingface.co/AUTOMATIC/stable-diffusion-3-medium-text-encoders/resolve/main/clip_l.safetensors"
27+
CLIPL_URL = f"{shared.hf_endpoint}/AUTOMATIC/stable-diffusion-3-medium-text-encoders/resolve/main/clip_l.safetensors"
2828
CLIPL_CONFIG = {
2929
"hidden_act": "quick_gelu",
3030
"hidden_size": 768,
@@ -33,7 +33,7 @@ def __getitem__(self, key):
3333
"num_hidden_layers": 12,
3434
}
3535

36-
CLIPG_URL = "https://huggingface.co/AUTOMATIC/stable-diffusion-3-medium-text-encoders/resolve/main/clip_g.safetensors"
36+
CLIPG_URL = f"{shared.hf_endpoint}/AUTOMATIC/stable-diffusion-3-medium-text-encoders/resolve/main/clip_g.safetensors"
3737
CLIPG_CONFIG = {
3838
"hidden_act": "gelu",
3939
"hidden_size": 1280,
@@ -43,7 +43,7 @@ def __getitem__(self, key):
4343
"textual_inversion_key": "clip_g",
4444
}
4545

46-
T5_URL = "https://huggingface.co/AUTOMATIC/stable-diffusion-3-medium-text-encoders/resolve/main/t5xxl_fp16.safetensors"
46+
T5_URL = f"{shared.hf_endpoint}/AUTOMATIC/stable-diffusion-3-medium-text-encoders/resolve/main/t5xxl_fp16.safetensors"
4747
T5_CONFIG = {
4848
"d_ff": 10240,
4949
"d_model": 4096,

modules/sd_disable_initialization.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -76,7 +76,7 @@ def transformers_modeling_utils_load_pretrained_model(*args, **kwargs):
7676
def transformers_utils_hub_get_file_from_cache(original, url, *args, **kwargs):
7777

7878
# this file is always 404, prevent making request
79-
if url == 'https://huggingface.co/openai/clip-vit-large-patch14/resolve/main/added_tokens.json' or url == 'openai/clip-vit-large-patch14' and args[0] == 'added_tokens.json':
79+
if url == f'{shared.hf_endpoint}/openai/clip-vit-large-patch14/resolve/main/added_tokens.json' or url == 'openai/clip-vit-large-patch14' and args[0] == 'added_tokens.json':
8080
return None
8181

8282
try:

0 commit comments

Comments
 (0)