Update misc.py
repair
This commit is contained in:
@@ -49,10 +49,12 @@ def instantiate_from_config(config, **kwargs):
|
|||||||
cls = get_obj_from_str(config["target"])
|
cls = get_obj_from_str(config["target"])
|
||||||
|
|
||||||
if config.get("from_pretrained", None):
|
if config.get("from_pretrained", None):
|
||||||
|
params_kwargs = config.get("params", {})
|
||||||
return cls.from_pretrained(
|
return cls.from_pretrained(
|
||||||
config["from_pretrained"],
|
config["from_pretrained"],
|
||||||
use_safetensors=config.get('use_safetensors', False),
|
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 = config.get("params", dict())
|
||||||
# params.update(kwargs)
|
# params.update(kwargs)
|
||||||
|
|||||||
Reference in New Issue
Block a user