power88/webui-fooocus-prompt-expansion
1
0
Fork 0
mirror of https://github.com/power88/webui-fooocus-prompt-expansion.git synced 2026-05-01 11:41:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix models_path

Browse Source
This commit is contained in:
w-e-w
2024-06-09 17:31:04 +09:00
parent 7486e593a5
commit 979e5e69da
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
scripts/expansion.py
View File

@@ -15,7 +15,7 @@ import psutil
from modules.scripts import basedir from modules.scripts import basedir
from transformers.generation.logits_process import LogitsProcessorList from transformers.generation.logits_process import LogitsProcessorList
from transformers import AutoTokenizer, AutoModelForCausalLM, set_seed from transformers import AutoTokenizer, AutoModelForCausalLM, set_seed
from modules import scripts, shared, script_callbacks from modules import scripts, paths_internal, shared, script_callbacks
from modules.ui_components import FormRow, FormColumn, FormGroup, ToolButton from modules.ui_components import FormRow, FormColumn, FormGroup, ToolButton
@@ -58,7 +58,7 @@ def get_free_memory(dev=None, torch_free_too=False):
SEED_LIMIT_NUMPY = 2 ** 32 SEED_LIMIT_NUMPY = 2 ** 32
neg_inf = - 8192.0 neg_inf = - 8192.0
ext_dir = basedir() ext_dir = basedir()
path_fooocus_expansion = os.path.join('.', "models", "prompt_expansion") path_fooocus_expansion = os.path.join(paths_internal.models_path, "prompt_expansion")
def safe_str(x): def safe_str(x):