Mirrors/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2025-12-28 13:42:02 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

add negative to positive

Browse Source
This commit is contained in:
maybleMyers 2025-12-05 03:55:20 -08:00
parent b91058b5b9
commit ecd8d4e66d
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
modules/ui.py
View File

@ -347,10 +347,17 @@ def expand_prompt_standalone(prompt: str, model_path: str, system_prompt: str =
log_step(" Loading model weights (this may take a moment)...")
model_start = time.time()
# Get the device the main program is using to avoid loading on wrong GPU
from backend import memory_management
main_device = memory_management.get_torch_device()
device_index = main_device.index if hasattr(main_device, 'index') and main_device.index is not None else 0
log_step(f" Target device: cuda:{device_index}")
_expansion_model_cache['model'] = Qwen3VLForConditionalGeneration.from_pretrained(
model_path,
torch_dtype=torch.bfloat16,
device_map="auto",
device_map={"": f"cuda:{device_index}"},
)
_expansion_model_cache['model_path'] = model_path
log_step(" Model weights loaded", model_start)