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

SDNQ fix model_ oader

Browse Source
This commit is contained in:
Disty0
2025-11-27 14:51:45 +03:00
parent ed6f977218
commit 131c51918b
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/sdnq/loader.py
+4
View File
@@ -103,6 +103,10 @@ def load_sdnq_model(model_path: str, model_cls: ModelMixin = None, file_name: st
quantization_config.pop("return_device", None)
quantization_config.pop("non_blocking", None)
quantization_config.pop("add_skip_keys", None)
quantization_config.pop("use_static_quantization", None)
quantization_config.pop("use_stochastic_rounding", None)
quantization_config.pop("use_grad_ckpt", None)
quantization_config.pop("is_training", None)
if hasattr(model_cls, "load_config") and hasattr(model_cls, "from_config"):
config = model_cls.load_config(model_path)