Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-18 08:44:33 +02:00
Code Issues Packages Projects Releases Wiki Activity

Pass torch_dtype to sdnq loader

Browse Source
This commit is contained in:
Disty0
2025-11-27 18:37:35 +03:00
parent 7b2a8e3f87
commit 73e4d1e379
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/sdnq/loader.py
+1 -1
View File
@@ -117,7 +117,7 @@ def load_sdnq_model(model_path: str, model_cls: ModelMixin = None, file_name: st
else:
model = model_cls(**model_config)
model = sdnq_post_load_quant(model, add_skip_keys=False, **quantization_config)
model = sdnq_post_load_quant(model, torch_dtype=dtype, add_skip_keys=False, **quantization_config)
key_mapping = getattr(model, "_checkpoint_conversion_mapping", None)
files = []