oobabooga/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2025-12-06 07:12:10 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Use dtype instead of torch_dtype

Browse source
This commit is contained in:
oobabooga 2025-12-02 06:35:06 -08:00
parent 5382fe2d1b
commit f4b27a39af
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
modules/image_models.py
View file

@ -120,7 +120,7 @@ def load_image_model(model_name, dtype='bfloat16', attn_backend='sdpa', cpu_offl
# Load the pipeline # Load the pipeline
load_kwargs = { load_kwargs = {
"torch_dtype": target_dtype, "dtype": target_dtype,
"low_cpu_mem_usage": True, "low_cpu_mem_usage": True,
} }