nodes/city96-ComfyUI_ExtraModels
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Allow changing T5 in extra_model_paths

Browse Source
This commit is contained in:
City
2024-05-06 16:36:38 +02:00
parent 47758b848b
commit 77d92abf1c
2 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
README.md
+2
View File
@@ -136,6 +136,8 @@ For faster loading/smaller file sizes, you may pick one of the following alterna
- [FP16 converted version](https://huggingface.co/theunlikely/t5-v1_1-xxl-fp16/tree/main) - Same layout as the original, download both safetensor files as well as the `*.index.json` and `config.json` files. - [FP16 converted version](https://huggingface.co/theunlikely/t5-v1_1-xxl-fp16/tree/main) - Same layout as the original, download both safetensor files as well as the `*.index.json` and `config.json` files.
- [BF16 converter version](https://huggingface.co/city96/t5-v1_1-xxl-encoder-bf16/tree/main) - Merged into a single safetensor, only `model.safetensors` (+`config.json` for folder mode) are reqired. - [BF16 converter version](https://huggingface.co/city96/t5-v1_1-xxl-encoder-bf16/tree/main) - Merged into a single safetensor, only `model.safetensors` (+`config.json` for folder mode) are reqired.
To move T5 to a different drive/folder, do the same as you would when moving checkpoints, but add ` t5: t5` to `extra_model_paths.yaml` and create a directory called "t5" in the alternate path specified in the `base_path` variable.
### Usage ### Usage
Loaded onto the CPU, it'll use about 22GBs of system RAM. Depending on which weights you use, it might use slightly more during loading. Loaded onto the CPU, it'll use about 22GBs of system RAM. Depending on which weights you use, it might use slightly more during loading.
T5/nodes.py
+4 -2
View File
@@ -7,13 +7,15 @@ from .loader import load_t5
from ..utils.dtype import string_to_dtype from ..utils.dtype import string_to_dtype
# initialize custom folder path # initialize custom folder path
# TODO: integrate with `extra_model_paths.yaml`
os.makedirs( os.makedirs(
os.path.join(folder_paths.models_dir,"t5"), os.path.join(folder_paths.models_dir,"t5"),
exist_ok = True, exist_ok = True,
) )
folder_paths.folder_names_and_paths["t5"] = ( folder_paths.folder_names_and_paths["t5"] = (
[os.path.join(folder_paths.models_dir,"t5")], [
os.path.join(folder_paths.models_dir,"t5"),
*folder_paths.folder_names_and_paths.get("t5", [[],set()])[0]
],
folder_paths.supported_pt_extensions folder_paths.supported_pt_extensions
) )