Update misc.py

repair
2025-08-06 01:14:49 +08:00
parent d9fc4d31bf
commit b3dd50ba37
1 changed files with 3 additions and 1 deletions
--- a/hy3dshape/hy3dshape/utils/misc.py
+++ b/hy3dshape/hy3dshape/utils/misc.py
@@ -49,10 +49,12 @@ def instantiate_from_config(config, **kwargs):
    cls = get_obj_from_str(config["target"])

    if config.get("from_pretrained", None):
+        params_kwargs = config.get("params", {})
        return cls.from_pretrained(
                    config["from_pretrained"], 
                    use_safetensors=config.get('use_safetensors', False),
-                    variant=config.get('variant', 'fp16'))
+                    variant=config.get('variant', 'fp16'),
+                    **params_kwargs) 

    params = config.get("params", dict())
    # params.update(kwargs)