Spaces:
Running
on
Zero
Running
on
Zero
Upload folder using huggingface_hub
Browse files
modules/SD15/SDToken.py
CHANGED
@@ -237,7 +237,7 @@ class SDTokenizer:
|
|
237 |
- `min_length` (int, optional): The minimum length of the input. Defaults to None.
|
238 |
"""
|
239 |
if tokenizer_path is None:
|
240 |
-
tokenizer_path = "_internal/sd1_tokenizer/"
|
241 |
self.tokenizer = tokenizer_class.from_pretrained(tokenizer_path)
|
242 |
self.max_length = max_length
|
243 |
self.min_length = min_length
|
|
|
237 |
- `min_length` (int, optional): The minimum length of the input. Defaults to None.
|
238 |
"""
|
239 |
if tokenizer_path is None:
|
240 |
+
tokenizer_path = "./_internal/sd1_tokenizer/"
|
241 |
self.tokenizer = tokenizer_class.from_pretrained(tokenizer_path)
|
242 |
self.max_length = max_length
|
243 |
self.min_length = min_length
|
modules/UltimateSDUpscale/USDU_upscaler.py
CHANGED
@@ -37,7 +37,7 @@ class UpscaleModelLoader:
|
|
37 |
#### Returns:
|
38 |
- `tuple`: The loaded model.
|
39 |
"""
|
40 |
-
model_path = f"_internal/ESRGAN/{model_name}"
|
41 |
sd = util.load_torch_file(model_path, safe_load=True)
|
42 |
if "module.layers.0.residual_group.blocks.0.norm1.weight" in sd:
|
43 |
sd = util.state_dict_prefix_replace(sd, {"module.": ""})
|
|
|
37 |
#### Returns:
|
38 |
- `tuple`: The loaded model.
|
39 |
"""
|
40 |
+
model_path = f"./_internal/ESRGAN/{model_name}"
|
41 |
sd = util.load_torch_file(model_path, safe_load=True)
|
42 |
if "module.layers.0.residual_group.blocks.0.norm1.weight" in sd:
|
43 |
sd = util.state_dict_prefix_replace(sd, {"module.": ""})
|